给你一个整数数组 nums
和一个整数 k
,请你统计并返回 该数组中和为 k
的子数组的个数。
子数组是数组中元素的连续非空序列。
示例 1:
输入:nums = [1,1,1], k = 2
输出:2
示例 2:
输入:nums = [1,2,3], k = 3
输出:2
cpp
class Solution {
public:
int subarraySum(vector<int>& nums, int k) {
unordered_map<int,int> f = {{0,1}};
int res = 0;
for(int i = 0, s = 0;i < nums.size(); i ++){
s += nums[i];
res += f[s - k];
f[s] ++;
}
return res;
}
};