leetcode 560. 和为 K 的子数组 （前缀和+哈希表）

class Solution {
public:
    //前缀和+哈希表
    //如果不用哈希表复杂度达到O(n2)超时
    //sums[i]表示从0到i-1的前缀和, sums[0] = 0, 因此j到i的连续和等于sums[i + 1] - sum[j]
    //求从j到i的连续和等于k的个数（j小于i，则sums[i + 1]一定大于sums[j]） 
    //<==> 求sums[i + 1] - sums[j] == k 的个数 
    //<==> 求 sums[j] == sum[i + 1] - k 的个数
    int subarraySum(vector& nums, int k) {
        int sz = nums.size();
        vector sums(sz + 1);
        mapmp;
        int ans = 0;
        sums[0] = 0; 
        mp[0] = 1;
        for(int i = 0; i < sz; i++){
            sums[i + 1] = sums[i] + nums[i];
            if(mp.find(sums[i + 1] - k) != mp.end()){
                ans += mp[sums[i + 1] - k];
            }
            mp[sums[i + 1]]++;
        }
        return ans;
    }
};