LeetCode题解2020/5/15 和为K的子数组

LeetCode 560

给定一个整数数组和一个整数 k，你需要找到该数组中和为 k 的连续的子数组的个数。

思路：

前缀和，所求的是一个连续的子数组，我们在对题目给定数组从前向后扫描一遍的同时，用一个变量Sum记录从头到扫描到该数的总和（也就是前缀和），用一个map<int,int>来存储前缀和和该前缀和出现的次数，若前缀和已经出现则次数加1，若没有出现则将其添加到map中。注意，初始化时要默认加入一个前缀和为0的项进入map，否则有可能错失数组长度为1的情况。

每扫描到一个新数，我们就在保存前缀和的map有没有sum-k的数，有多少这样的前缀和，就代表有多少以该数结尾的连续的数组符合题目的要求，就把他添加到我们的题目结果中。这样我们仅需扫描数组一遍即可获得结果。

代码：

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