一、题目
给你一个整数数组 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
二、题解
class Solution {
public:
int subarraySum(vector<int>& nums, int k) {
int n = nums.size();
//记录前缀和为key的子数组的数量
unordered_map<int,int> map;
map[0] = 1;
int res = 0,sum = 0;
for(int i = 0;i < n;i++){
sum += nums[i];
//map[sum-k]表示在i位置之前,有多少前缀和为sum-k的子数组
res += map[sum - k];
map[sum]++;
}
return res;
}
};