solution.py

# Будем юзать динамическое программирование - храним мапу суммы элементов до текущего элемента.

# Два случая:
# 1. Либо подмассив с элементами от начала исходного массива до 
# текущего элемента имеет сумму элементов равную К.
# 2. Либо существует более короткий подмассив (от некоторого 
# среднего элемента) до этого элемента, и в этом случае разница 
# между общей суммой до среднего элемента и общей суммой до этого элемента равна k.

from collections import defaultdict

def subarray_sum_to_k(nums, k):
    dp, total, count = defaultdict(int, {0: 1}), 0, 0
    
    for num in nums:
        total += num
        diff = total - k
        count += dp[diff]

        dp[total] = dp[total] + 1

    return count