题目: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;
}
};