数组前缀和

前缀和

前缀和就是指前缀的和，例如在数组中，从开始到 i 就是到 i 的前缀和。前缀和一般用来求中间连续某一段的和，例如sumi - sumj - 1就可以求出j 到 i 这一段的和。

在这一道题目里面，中间某一段连续子数组和为k，意思即为sumi - sumj - 1 = k，也即sumi - k = sumj - 1，所以当我们求每一个的前缀和的时候，只需要统计对应的sumj - 1个数即可。sum对应的个数用map存储。注意map0 = 1，即本身本来有一个。

class Solution {
public:
    int subarraySum(vector<int>& nums, int k) {
        int ans = 0;
        unordered_map<int, int> mp;
        int pre = 0;
        mp[0] = 1;
        for (int i : nums) {
            pre += i;
            ans += mp[pre - k];
            ++mp[pre];
        }
        return ans;
    }
};

当然还有二维数组的前缀和，不过大致原理就是这样，通过两个前缀和求中间的和。