560. 和为K的子数组
1.题目描述
给定一个整数数组和一个整数 k,你需要找到该数组中和为 k 的连续的子数组的个数。
示例 1 :
说明 :
1.数组的长度为 [1, 20,000]。
2.数组中元素的范围是 [-1000, 1000] ,且整数 k 的范围是 [-1e7, 1e7]。
2.思路
如果累计总和(sum[i]表示累加到下标i的和),在索引 i 和 j 处相差 k,即 sum[i] - sum[j] = k,则位于索引 i 和 j 之间的元素之和是 k。
基于这个想法,可以使用了一个哈希表 map,它用于存储所有可能的索引的累积总和以及相同累加和发生的次数。存储数据形式:(sum_i,sum_i的出现次数)。遍历数组,记录每个累积和的次数,判断sum_i - k是否存在map中,如果存在,结果次数count累加
3.代码
class Solution {
public:
int subarraySum(vector<int>& nums, int k) {
unordered_map<int,int> m;
m[0] = 1;
int sum = 0;
int count = 0;
for(int i = 0;i < nums.size();++i){
sum += nums[i];
if(m.find(sum - k) != m.end()){
count += m[sum-k];
}
m[sum]++;
}
return count;
}
};
4.复杂度分析
时间复杂度:O(n)
空间复杂度:O(n)