(nice!!!)LeetCode 560. 和为 K 的子数组（哈希表、前缀和）

题目：560. 和为 K 的子数组

思路：找出所有和为k的子数组，这里可以想到用前缀和。然后用一层循环来枚举子数组的右端点r。左端点l就通过哈希表来寻找。如果区间[l,r]的和为k，即sum[r]-sum[l-1]==k，那在遍历到r时，我们只需要查找是否有sum[r]-k的前缀和出现过。同时当sum[r]==k时，表示[0,r]也是符合要求的。细节看注释

class Solution {
public:
    int subarraySum(vector<int>& nums, int k) {
    	//前缀和
        for(int i=1;i<nums.size();i++){
            nums[i]+=nums[i-1];
        }
        //用于保存已经遍历过的前缀和，已经出现的次数
        unordered_map<int,int> mp;
        //答案
        int ans=0;
        //用一层循环来枚举子数组的右端点r
        for(auto x:nums){
        	//左端点l就通过哈希表来寻找。如果区间[l,r]的和为k，即sum[r]-sum[l-1]==k，
        	//那在遍历到r时，我们只需要查找是否有sum[r]-k的前缀和出现过。
            if(mp.find(x-k)!=mp.end()){
                ans+=mp[x-k];
            }
            //保存当前的前缀和
            mp[x]++;
            //当sum[r]==k时，表示[0,r]也是符合要求的。
            if(x==k) ans++;
        }
        return ans;
    }
};