给你一个整数数组 nums
和一个整数 k
,请你统计并返回 该数组中和为 k
的子数组的个数 。
子数组是数组中元素的连续非空序列。
示例 1:
输入:nums = [1,1,1], k = 2 输出:2
示例 2:
输入:nums = [1,2,3], k = 3 输出:2
思路
使用前缀和+哈希,计算数组和并存入哈希,在哈希中查找是否存在k-sum的键。
代码
class Solution {
public:
int subarraySum(vector<int>& nums, int k) {
int size = nums.size();
int res = 0,sum = 0;
unordered_map<int, int> hashtable;
hashtable[0] = 1;
for(int i = 0; i < size; ++i)
{
sum+=nums[i];
if(hashtable.find(sum - k) != hashtable.end())
res += hashtable[sum - k];
hashtable[sum]++;
}
return res;
}
};