LeetCode-560-和为K的子数组

1、前缀和+哈希表

为了减少不必要的计算区间和的过程，我们可以使用前缀和来代替数组。前缀和preSum[i]表示nums数组中前i个元素之和。当我们需要计算区间[i,j]和时，只需要preSum[j]-preSum[i-1]即可，这样做能够大大减少不必要的相加。

同时，为了避免我们在前缀和数组中循环过多，我们采用哈希表进一步加快查询速度。我们在每次循环中先计算出当前i个元素之和，而后在哈希表中寻找是否有sum - k的键值对。若有说明二者的区间之和为k，将preSum[sum - k]的值加到result上面。

class Solution {
public:
    int subarraySum(vector<int> &nums, int k) {
        unordered_map<int, int> preSum;
        int sum = 0, result = 0;
        preSum[0] = 1;
        for (int i: nums) {
            sum += i;
            if (preSum.find(sum - k) != preSum.end()) result += preSum[sum - k];
            ++preSum[sum];
        }
        return result;
    }
};