560. 和为K的子数组(一般)

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

示例 1 :

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

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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/subarray-sum-equals-k

思路：

前缀和

代码1：

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
class Solution {
    public int subarraySum(int[] nums, int k) {
        //前缀和
        int len = nums.length;
        if(len==0){
            return 0;
        }
        int[] presum = new int[len+1];
        presum[0]=0;
        for(int i=0;i<len;i++){
            presum[i+1] = presum[i]+nums[i];
        }
        int count = 0;
        for(int i = 0;i<len;i++){
            for(int j=i;j<len;j++){
                if(presum[j+1]-presum[i]==k){
                    count++;
                }
            }
        }
        return count;
    }
}

代码2

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
class Solution {
    public int subarraySum(int[] nums, int k) {
        //前缀和,使用hash优化
        //当前数之前，有多少前缀和等于当前的【前缀和-k】
        HashMap<Integer,Integer> map = new HashMap<>();
        //表示下表为0的元素前面没有元素，可以认为前缀和为0，个数为1个，
        map.put(0,1);
        int presum = 0,count=0;
        for (int num:nums){
            presum += num;
            if(map.containsKey(presum-k)){
                count+=map.get(presum-k);
            }
            map.put(presum,map.getOrDefault(presum,0)+1);
        }
        return count;
    }
}