【做一道算一道】和为 K 的子数组

给你一个整数数组 nums 和一个整数 k ，请你统计并返回 该数组中和为 k 的子数组的个数 。

子数组是数组中元素的连续非空序列。

示例 1：

输入：nums = [1,1,1], k = 2
输出：2
示例 2：

输入：nums = [1,2,3], k = 3
输出：2

提示：

1 <= nums.length <= 2 * 104
-1000 <= nums[i] <= 1000
-107 <= k <= 107

思路

看到想到是滑动窗口，调了力扣结果和本地调的对不上，看题解，发现说有负值，那滑动就不行。
官解是前缀和，记一下。
其中关键代码是：

count += mp[pre - k]; // 如果存在前缀和为pre - k，更新count

表示如果mp中存在pre - k，说明存在一个前缀和为pre - k，那么从这个前缀和的末尾到当前索引的子数组和为k。即从pre - k前缀和的末尾到当前索引位置的子数组满足条件，可被记录。
因此，将count增加mp[pre - k]（该前缀和出现的次数，每个对应一种满足的情况）。

代码

滑动窗口（仅适合全为正）：
奇怪的点，全为正的时候，本地调是正确的，力扣上结果就不对，不知道哪儿有问题。

class Solution {
public:
    int subarraySum(vector<int>& nums, int k) {
        int sum=0;
        int l=0,r=0;
        int cnt=0;
        sort(nums.begin(),nums.end());
        if(k<nums[0])
        return cnt;

        while(r<nums.size()){
            sum+=nums[r++];

            while(sum>=k){
                if(sum==k){
                    cnt++;
                }
                sum-=nums[l++];  
            }
        }
        return cnt;
    }
};

前缀和：

class Solution {
public:
    int subarraySum(vector<int>& nums, int k) {
        unordered_map<int, int> mp;
        mp[0] = 1; // 初始化前缀和为0的出现次数为1
        int count = 0, pre = 0;
        for (auto x : nums) {
            pre += x; // 更新前缀和
            count += mp[pre - k]; // 如果存在前缀和为pre - k，更新count
            mp[pre]++; // 更新当前前缀和的出现次数
        }
        return count;
    }
};

官解当中多一个判定：

if (mp.find(pre - k) != mp.end())

可省略，因为即使 mp[pre - k] 之前没有出现过，其值也是0，不会对 count 产生影响。