前缀和

力扣题目

560. 和为K的子数组

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

示例 1 :

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

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

 

解法说明

该题对应题解已经很详细的说了常规思路和优化方法。不再重复。本文仅对前缀和+hash表解法做说明。

前缀和思路为：先计算出前面每项的和，当前面j个数的和减去k等于前面i个数的和时，则i+1到j的数就是满足条件的。

通过示例来说明：

比如我有如下几个数，给定K为3 

数队列： 

直观看，索引0:1的数12，索引1:2的数21，索引2：3的数12是满足条件的，故结果为3。

计算上面数队列的前缀和，结果为：

该问题转换为遍历每个前缀和，分两种情况：

1. 看它自己是否等于K，如果是，则说明从索引0到该位置的数满足要求。比如遍历到前缀和索引1处，其值为3，刚好与K相等，则索引0到索引1的数时满足要求的。
2. 该前缀和减去K的值与前面多少个前缀和相同，满足条件，则说明从其前面+1的位置到当前位置的数满足要求、比如K为3时，当遍历到前缀和索引2时，即值为4时，4-3=1，其与索引0的前缀和相等，则说明索引1到索引2的数是满足要求的

伪代码如下(假设已计算出前缀和并放到数组preSum中):

result = 0
for (int i = 0; i < lenth(preSum); i++) {
    if(preSum[i] == K) {
        // 这个是处理前面说的情况1
        result++;
    }
    // 处理前面说的情况2
    for (int j = 0; j < i; j++) {
        if (preSum[i] - K == preSum[j]) {
            result++;
        }
    }
}

其实我们并不关心哪到哪的数之和满足条件，我们只关心满足条件的这些数有多少种。对上面代码的第二个循环进行优化：

用一个map来存储前面已有的前缀和的个数，当当前前缀和减去K的值在map中存在时，则说明有N个满足情况的组合存在。但是这样没法处理前面说的情况1，也就是当当前前缀和就等于K时，没法处理。为了处理这种情况，我们在map中预先放入值为0的数，让其值等于1，这样就可以处理这个情况了。示例代码为：

map<int, int> infos;
infos[0] = 1;
result = 0;
for (int i = 0; i < length(preSum); i++) {
    int tmp = preSum[i] - k;
    auto itFind = infos.find(tmp);
    if (itFind != infos.end()) {
        result += itFind->second;
    }
    // 将当前sum放入到map中
    auto it = infos.find(preSum[i]);
    if (it != infos.end()) {
        it->second++;
    } else {
        infos[preSum[i]] = 1;
    }
}