高频面试题——和为 k 的子数组

剑指 Offer II 010. 和为 k 的子数组

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

示例 1：

输入:nums = [1,1,1], k = 2
输出: 2
解释: 此题 [1,1] 与 [1,1] 为两种不同的情况
示例 2：

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

提示:

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

思路

数组：0…i…j…end

当我们去想要判断第j个数是否有成立的条件时，我们只需要去判断第j个数减去0到i个数的和（0<i<j）是否等于k即可判断有无成立条件（区间i+1到j的和）。

所以我们只要吧前i个数的和记录下来，作为前缀，然后再用查找效率相对较高的hashmap进行记录前缀，即可快速找出答案。

class Solution {
    public int subarraySum(int[] nums, int k) {
        HashMap<Integer,Integer> map = new HashMap<>(); 
        int pre = 0,count = 0;
        map.put(0,1);
        for(int i=0;i<nums.length;i++){
            pre += nums[i];   
            if(map.containsKey(pre-k)){
                count += map.get(pre-k);
            }
            //判断是否存在per键，若存在，则取pre值+1，若不存在，则创建一个，值为0
            map.put(pre,map.getOrDefault(pre,0)+1);   
        }
        return count;
    }
}

代码为官方答案，此处记录个人理解思路