leetcode刷题（54）——560. 和为K的子数组

一、题目

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

示例 1 :

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

说明 :

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

二、思路——哈希表

我们定义 sum[i] 表示数组前 i 项和，即

• sum[0] = nums[0]
• sum[1] = nums[0] + nums[1]
• sum[2] = nums[0] + nums[1] + nums[2]
  … …

sum 之间做减法，有

• sum[1] - sum[0] = nums[1]
• sum[2] - sum[0] = nums[1] + nums[2]

我们发现，sum[j] - sum[i] 表示区间 (i，j]（左开右闭） 内数组元素的和。如果累计和在索引 i 和 j 处相差 k，即 sum[j] - sum[i] = k，则位于索引 i 和 j 之间的元素之和是 k。

所以，我们可以使用哈希表存储所有前 i 项的累计和及其出现的次数。遍历数组 nums 并寻找累计和，

• 如果遇到一个新的和，在 hashmap 中创建一个与该总和相对应的新条目；
• 如果再次出现相同的和，我们增加与 map 中的和相对应的计数；
• 此外，对于遇到的每个总和 sum[i]，如果 sum[i] - k 等于 map 中的某个 sum，说明该 sum 和 sum[i] 之间元素的和为 k，则将该 sum 对应的出现次数加给 count。

三、代码

class Solution {
    public int subarraySum(int[] nums, int k) {
        int count = 0;
        HashMap<Integer, Integer> map = new HashMap<>();
        map.put(0, 1);
        int sum = 0;
        for(int i = 0; i < nums.length; i++){
            sum += nums[i];
            if(map.containsKey(sum - k))
                count += map.get(sum - k);
            if(map.containsKey(sum))// 已有和
                map.put(sum, map.get(sum) + 1);
            else// 新的和
                map.put(sum, 1);
        }
        return count;
    }
}