给你一个整数数组 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;
}
};
标签:子串,前缀,560,sum,int,mp,数组,ans
From: https://www.cnblogs.com/lihaoxiang/p/17577138.html