力扣-560.和为K的子数组
题目描述:给你一个整数数组 nums 和一个整数 k ,请你统计并返回 该数组中和为 k 的子数组的个数 。
子数组是数组中元素的连续非空序列。
方法1:暴力枚举
两层for循环嵌套,或者使用递归算法,不赘述。
方法2:前缀表
看了一眼力扣官方题解,一堆公式和繁杂的解释,实在听不下去,干脆自己写个通俗易懂的说明。
官方题解:
↑↑↑↑↑↑↑话多,懒得看↑↑↑↑↑↑↑↑↑↑
方法2围绕一个问题:为什么?
1、为什么可以使用空间换时间?
2、为什么只用一遍循环?
第一遍循环计算无需多言,正常for循环即可遍历计算出黑色框住的sum和;
当以第n(图中为第3)个节点开始循环时,若黑色框选范围满足sum==k,则紫色框选范围满足:sum(1-2) = sum(1-5)+k,而sum(1-5)在第一次循环中已经计算出,由此发现可以通过第一次循环的记录得到所有不以第一个节点起始的,满足和为k的子序列
即可得到以下代码:
class Solution {
public:
int subarraySum(vector<int>& nums, int k) {
unordered_map<int,int> mp;//记录第一次遍历的sum
mp[0] = 1;
int result = 0;
int pre = 0;
for(auto&x :nums){
pre+=x;
if(mp.find(pre-k)!=mp.end()){//如果找到
result += mp[pre-k];
}
mp[pre]++;
}
return result;
}
};