题目描述:
给你一个整数数组?nums
?和一个整数?k
?,请你统计并返回?该数组中和为?k
?的子数组的个数?。
子数组是数组中元素的连续非空序列。
示例 1:
输入:nums = [1,1,1], k = 2 输出:2
示例 2:
输入:nums = [1,2,3], k = 3 输出:2
提示:
1 <= nums.length <= 2 * 104
-1000 <= nums[i] <= 1000
-107 <= k <= 107
思路:遍历数组使用一个变量presum记录当前节点的前缀和,用另一个指针去遍历该位置前面的数组,找到和为k的子数组,但是这样的话时间复杂度就是o(n^2)
优化:采用哈希表记录前缀和,把找和为k的子数组转化为在哈希表中找和为presum-k的前缀和数组,如果哈希表中存在则将结果ret加上该前缀和所对应的哈希值(即出现的次数),再将该位置的前缀和放入哈希表中。注意特殊情况,如果presum==k则将ret++。
代码实现:
class Solution {
public:
int subarraySum(vector<int>& nums, int k) {
int presum=0,ret=0;
unordered_map<int,int> hash;
for(auto e: nums)
{
presum+=e;
if(hash.count(presum-k))
ret+=hash[presum-k];
if(presum==k)
ret++;
hash[presum]++;
}
return ret;
}
};