给你一个整数数组 nums
和一个整数 k
,请你统计并返回该数组中和为 k
的连续子数组的个数。
--------------------------------------------------------------------------------------------------------------------------------
方法一:如果不要求时间,最简单易懂得方法就是将所有数据从头到尾遍历一遍,再对每个数字的前面或者后面的数组进行遍历,即可得结果。
方法二:
我们定义 pre[i] 为 [0..i]里所有数的和,则pre[i] 可以由 pre[i−1] 递推而来:
pre[i]=pre[i−1]+nums[i]
进而可知满足以i结尾和为k的连续子数组一定满足:
pre[j−1]=pre[i]−k
当然这里的j-1可以直接用j表示,代表的意思是在i之前存在有前缀等于pre[i]−k的位置。
class Solution {
public:
int subarraySum(vector<int>& nums, int k) {
unordered_map<int,int>mp;//建立哈希表,键表示前缀值,值表示的是这个前缀值在数组从前到后遍历的过程中出现的次数。
mp[0]=1;//建立一个前缀为0的初始值,为的是统计出从第一个数开始的满足条件的子数组
int count=0,pre=0;//count统计的是满足条件的子数组的总数,
for(auto&x:nums)//这里面的x就是nums中的每个数,在循环中每次出现一个
{
pre+=x;
if(mp.find(pre-k)!=mp.end())//首先mp.end()在mp最后一个位置的后面,与其相等说明遍历所有也没找到目标值
{
count+=mp[pre-k];
}
++mp[pre];//将统计到的pre保存到mp中
}
return count;
}
};