剑指 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 * 10⁴
• -1000 <= nums[i] <= 1000
• -10⁷ <= k <= 10⁷

思路

1.要找出连续子数组个数就需要枚举每个元素
2.题目要求子数组的和满足整数k，这时可以用前缀和求和，任意子数组的和可以用从下标为0到下标为子数组右端点的数组之和-从下标为0到下标为子数组左端点的数组之和，也就是，我们需要从头到每个元素的子数组之和，这可以在枚举元素累加元素求
3.若枚举到下标为i的元素时，并且前缀和为x，那么只需要找出有多少个j（j<i）使得前缀和为x-k，就可以得到和为k的子数组个数
4.为了能够快速查找到前缀和为x-k的子数组个数有多少，我们可以用哈希表记录，key为前缀和，value为出现次数

代码

class Solution {
    public int subarraySum(int[] nums, int k) {
        Map<Integer,Integer> map=new HashMap<Integer,Integer>();
        map.put(0,1);
        //若nums[i]等于k，则sum-k==0，那么自身也该被计入cnt里
        int cnt=0;
        int sum=0;
        for(int num:nums){
            sum+=num;
            cnt+=map.getOrDefault(sum-k,0);
            map.put(sum,1+map.getOrDefault(sum,0));
        }
        return cnt;
    }
}