leetcode中等560和为k的子数组

Posted 2021-06-06 qq_40707462

思路：
[j,i]这个子数组和为 k 这个条件可以转化为sum[i]−sum[j−1]==k

使用hash表存储子串和
key：0到当前位置的子串和
value：该和出现的次数
（比如[-1,1,0]，和为0出现了2次）

每遍历到nums[i]，如果sum[i]-k的结果出现在hash里，表示存在[j,i]使得和为k

class Solution:
    def subarraySum(self, nums: List[int], k: int) -> int:
        hash={}
        count=0
        sum=0
        for i in range(len(nums)):
            sum+=nums[i]
            if sum==k:#[0,i]和为k
                count+=1
            if (sum-k) in hash:#[i,j]和为k，i可能有多个
                count+=hash[sum-k]
            hash[sum]=hash.get(sum,0)+1
        return count