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

前缀和数组

数组nums =[1 2 3 4 5]
前5项和 s5 = 1+2+3+4+5 = 15
前2项和 s2 = 1+2 = 3
则3-5项的和 = s5-s2 = 12
则2-3项的和 = s3-s1 = 5
上面的数组 可以生成对应的前缀和数组
前缀和数组 第一个元素 需要为0 这样通过前缀和数组 元素的相减 才能获取原数组指定下标之间的和
s = [0,1,3,6,10,15]
例如 获取nums中[1,2]
s[3] - s[1] = 5
获取nums中[0,2]
s[2] - s[0] = 3

本解法需要解释的地方

定义一个map 来存放原数组对应的前缀和数组 key = 前缀和数组元素 val = 该元素出现的次数

map.put(0,1) //前缀和数组 第一个元素为0

//k为目标值 sum - k =  sum-k
/**
例子
sum = 10 k = 4 sum-k = 6

假定一个数组 满足sum = 10 k = 4 sum-k = 6

1 2 3 4
其前缀和数组为
[0 1 3 6 10]

sum = s[4] 前4项和为10 k=4 sum-k = 6 = s[3]前3项和为6 
能从前缀和数组中找到sum-k 表明
s[3]
-----
1 2 3 4
-------
s[4]

k = s[4]-s[3] 根据前缀和数组 元素之间相减 可以获取到原数组 指定下标之间的元素值 的特点 我们可以知道 k是存在的

*/
if(map.containsKey(sum-k)){
	ans += map.get(sum-k);
}

class Solution {
    public int subarraySum(int[] nums, int k) {
        Map<Integer, Integer> map = new HashMap<>();
        map.put(0, 1);
        int sum = 0;
        int ans = 0;
        for (int i = 0; i < nums.length; i++) {
            sum += nums[i];
            if (map.containsKey(sum - k)) {
                ans += map.get(sum - k);
            }
            map.put(sum, map.getOrDefault(sum, 0) + 1);
        }

        return ans;
    }
}