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

题目

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

思路

这次数组里有负数了，所以之前的双指针不能用了。

直接的暴力解法自然是可以穷举出所有的连续子数组，然后找到符合条件的即可，但是略微复杂。

这里可以遍历一次数组，将从第0位到第n位的加和保存成新的整数数组，这样我们就可以通过相减来快速获取任意一个连续子数组的加和。

解法一：循环遍历计算加和

代码

public int subarraySum(int[] nums, int k) {
    int[] sums = new int[nums.length + 1];
    int sum = 0;
    for (int i = 0; i < nums.length; i++) {
        sum += nums[i];
        sums[i + 1] = sum;
    }

    int result = 0;
    for (int i = 0; i < nums.length; i++) {
        for (int j = i; j < nums.length; j++) {
            if (k == sums[j + 1] - sums[i]) {
                result++;
            }
        }
    }

    return result;
}

解法二：一次遍历

逻辑

再进一步优化。我们是不是可以在第一次遍历加和的过程中，就能得知截止到当前位置的元素，前面有几段连续的子数组可以满足加和等于目标k？

其实我们根本不需要知道，第0位到第n位的加和是多少。我们只需要知道，某个加和出现过多少次即可。

举个例子，比如k为6，数组为[2,3,-4,1,6,3,-2]。我们来分别计算sum(n)，sum(n)-k，如下图所示：

当我们遍历到下标为4的元素时，从0到4的数组加和为8。此时，sum(n)-k=2。在sum(n)数组中，有两个加和为2的情况。所以我们可以得知，子数组[3,-4,1,6]（下标1到下标4）和子数组[6]都是符合条件的。那么我们只要统计出来sum为2出现的次数即可。进一步讲，当我们遍历到一个元素时，我们只需要看看之前的sum(n)数组中存在几个值为sum(n)-k的元素即可。这里我们用一个map来维护这部分信息，key为sum值，value为出现的次数。

代码

public int subarraySum(int[] nums, int k) {
    // 用一个map记录某个sum出现的次数
    Map<Integer, Integer> sumCountMap = new HashMap<>();
    sumCountMap.put(0, 1);
    int result = 0;
    int sum = 0;
    for (int i = 0; i < nums.length; i++) {
        sum += nums[i];
        // 存在的次数
        result += sumCountMap.getOrDefault(sum - k, 0);
        // 本次的sum出现次数加一
        sumCountMap.put(sum, sumCountMap.getOrDefault(sum, 0) + 1);
    }
    return result;
}