560. 和为 K 的子数组（前缀和解决子串问题）

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

示例 1：

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

> 思路

• 每个元素对应一个“前缀和”
• 遍历数组，根据当前“前缀和”，在 map 中寻找「与之相减 == k」的历史前缀和
• 当前“前缀和”与历史前缀和，差分出一个子数组，该历史前缀和出现过 c 次，就表示当前项找到 c 个子数组求和等于 k。
• 遍历过程中，c 不断加给 ans，最后返回 ans

> 代码

class Solution {
public:
    int subarraySum(vector<int>& nums, int k) {
        int sum = 0, ans = 0;
        unordered_map<int,int> mp;
        mp[0] = 1;
        for(int i: nums){
            //找所有的前缀和
            sum += i;
            if(mp.find(sum-k) != mp.end()) ans += mp[sum-k];
            mp[sum] ++;
        }
        return ans;
    }
};