Участник:Ковальков Антон М05-903б/Subarray Sum Equals K

Материал из DISCOPAL
Перейти к: навигация, поиск

https://leetcode.com/problems/subarray-sum-equals-k/

Python3

from collections import defaultdict
 
class Solution:
    def subarraySum(self, nums: List[int], k: int) -> int:
        s = defaultdict(int)
        s[0] = 1
        p = 0
        c = 0
        for n in nums:
            p += n
            if p-k in s:
                c += s[p-k]
            s[p] += 1
        return c