题目描述

给你一个整数数组 nums 和一个整数 k ，请你统计并返回 该数组中和为 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) {
        int res = 0;
        for (int i = 0; i < nums.length; i++) {
            int sum = 0;
            for (int j = i; j < nums.length; j++) {
                sum+=nums[j];
                if (sum==k){
                    res++;
                }
            }
        }
        return res;
    }
}

前缀和加哈希

class Solution {
    public int subarraySum(int[] nums, int k) {
        //key:表示前缀和 value：有相同前缀和的个数
        Map<Integer,Integer> preSumFreq = new HashMap<>();
        //对于下标为0的元素，前缀和为0，个数为1
        preSumFreq.put(0,1);

        int preSum = 0;//表示前缀和
        int count = 0;
        for (int i = 0; i < nums.length; i++) {
            preSum = preSum+nums[i];
            //先得到前缀和为preSum-k的个数，然后加到计数变量里
            //具体的解释为：假设有个位置j，其中j<i，j的前缀和为preSum-k，
            // 因为当前位置的前缀和为preSum,所以证明从j到i的和为：preSum-(preSum-k)=k
            // 也就是我们想要求的子数组的和为k的情况
            if (preSumFreq.containsKey(preSum-k)){
                count+=preSumFreq.get(preSum-k);
            }
            //继续更新preSumFreq，保持统计前缀和
            preSumFreq.put(preSum,preSumFreq.getOrDefault(preSum,0)+1);
        }
        return count;
    }
}