LeetCode560. 和为K的子数组

1. 题目

给定一个整数数组和一个整数 k，你需要找到该数组中和为 k 的连续的子数组的个数。

示例 1 :

输入:nums = [1,1,1], k = 2
输出: 2 , [1,1] 与 [1,1] 为两种不同的情况。

说明 :

数组的长度为 [1, 20,000]。
数组中元素的范围是 [-1000, 1000] ，且整数 k 的范围是 [-1e7, 1e7]。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/subarray-sum-equals-k
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

2. 题解

用字典 sum_dict[val] = i,记录从左边元素，累加到第i个元素的和val;
然后，遍历val_i - val_j = k;且val_i的索引比val_j的索引大的个数。

代码

class Solution:
    def subarraySum(self, nums: List[int], k: int) -> int:
        # 找 val_i - val_j == k;i>j;
        # 放入dict中查找
        sum_dict = {}
        s = 0
        for i in range(len(nums)):
            s += nums[i]
            if s in sum_dict:
                sum_dict[s].append(i)
            else:
                sum_dict[s] = [i]
        ans = 0
        for val in sum_dict.keys():
            if val == k:
                ans += len(sum_dict[val])
            if (val-k) in sum_dict.keys():
                val_list = sum_dict[val]
                val_k_list = sum_dict[val-k]
                for i in val_list:
                    for j in val_k_list:
                        if i>j:
                            ans += 1
        return ans