LeetCode--HOT100题（10）

题目描述：560. 和为 K 的子数组（中等）

给你一个整数数组 nums 和一个整数 k ，请你统计并返回 该数组中和为 k 的连续子数组的个数 。

LeetCode做题链接：LeetCode-和为 K 的子数组

示例 1：

输入：nums = [1,1,1], k = 2
输出：2

示例 2：

输入：nums = [1,2,3], k = 3
输出：2

提示：

1 <= nums.length <= 2 * 104
-1000 <= nums[i] <= 1000
-107 <= k <= 107

题目接口

class Solution {
    public int subarraySum(int[] nums, int k) {

    }
}

解题思路

为什么是 count += mp.get(pre - k); 呢 ：
举个例子: k = 6, 数组 [1, 2, 3, 0, 6] 累加和为: [1, 3, 6, 6, 12], 明显答案应该是 4, 当我们来到第一个累加和为 6 的位置上时, pre - k = 0, 也就是说从下标 0 到当前位置的累加和是一个答案, 当来到第二个 6 的位置上时, 也就是说从下标 0 到当前位置的累加和是一个答案, 而当来到 12 位置上时, pre - k = 6, 也就是说从累加和为 6 的子数组的后一个位置到当前位置也是满足条件的答案, 而累加和为 6 的子数组不只有一个, 所以 count 是 加 mp.get(pre - k);, 而不是加 1。

代码

class Solution {
    public int subarraySum(int[] nums, int k) {
		int res = 0; // 统计结果
        int sum = 0;   // 前缀和
        HashMap<Integer, Integer> hashMap = new HashMap<>(); // 哈希表

        for (int num : nums) { // 从左往右
            sum += num; //累加
            if (sum == k) { // 从下标 0 累加到 i 刚好等于 k 的这一类
                res++;
            }
            // 若哈希表包含了(sum-k)这个键，则把 这个键对应的值加到结果中
            if (hashMap.containsKey(sum - k)) {
                res += hashMap.get(sum - k);
            }
            // 放入哈希表中，若存在，则值加一；若不存在，则存入（sum, 1）
            hashMap.put(sum, hashMap.getOrDefault(sum, 0) + 1);
        }
        return res;
    }
}

成功！

PS:

感谢您的阅读！如果您觉得本篇文章对您有所帮助，请给予博主一个赞喔~