中等
给你一个整数数组 nums
和一个整数 k
,请你统计并返回 该数组中和为 k
的子数组的个数 。
子数组是数组中元素的连续非空序列。
示例 1:
输入:nums = [1,1,1], k = 2 输出:2
示例 2:
输入:nums = [1,2,3], k = 3 输出:2
提示:
1 <= nums.length <= 2 * 10^4
-1000 <= nums[i] <= 1000
-10^7 <= k <= 10^7
思路:(忘了负数的情况,不完全对)使用滑动窗口的思想,first指向滑动窗口的起始下标,last指向滑动窗口的最后一个下标。初始状态均指向数组第一个下标:
(1)当sum>k时,说明窗口中总和过多,first向前移动,收缩窗口
(2)sum<k时,过小,last向后移动,扩大窗口
(3)sum=k时 first向前移(可能存在0的情况)
注意:每次移动first,last后要判断移动后是否还存在sum==k;
class Solution {
public:int subarraySum(vector<int>& nums, int k) {int first=0;int last=0;int sum=0;int times=0;//注意:不要溢出,负数,while(last<nums.size()){sum+=nums[last];while(sum==k) //{times++; sum-=nums[first];if(first+1<nums.size())first++;else break;}if(sum>k){while(sum>k){//first向前移,sum-nums[first] //1 1 0 0 1 1 3 0 0 4 (4)sum-=nums[first]; if(first+1<nums.size())first++;else break;//每移动一次也要判断是否等于while(sum==k){times++;sum-=nums[first]; if(first+1<nums.size())first++;else break;}}}if(last+1<nums.size())last++;else break;}return times;}
};
正确思路:使用前缀和+哈希表进行优化 (小红书社招二面出现过):
(1)首先使用sum[i]统计i小标之前的所有数之和(包括下标i指向的数),因此某一段序列和为k转化为sum[p]-sum[q]=k,;有效解决负数带来的问题。在遍历过程中用value记录前i个数组值得总和,使用哈希表记录,其中sum[value]对应的值表示和为value的数组段。
(2)遍历一遍sum,当前sum[p]已知,k已知,只要判断是否存在sum[q]=sum[p]-k,就能判断当前下标作为开始,是否存在一段数组之和等于k,出现次数times=times+sum[q];
class Solution {
public:int subarraySum(vector<int>& nums, int k) {int times=0;map<int,int>sum1;int sum[30000];for(int i=0;i<nums.size();i++){if(i==0){sum[0]=nums[0];sum1[sum[0]]++;}else{sum[i]=sum[i-1]+nums[i];sum1[sum[i]]++;}int need=sum[i]-k;//有一个问题:别把当前数算进去,需要-1,当前数之和也为-1if(sum1.find(need)!=sum1.end()){if(need==sum[i]){times+=(sum1[need]-1); //[-1,-1,1]中重复计算了}elsetimes+=sum1[need];}if (need==0){//注意【0,1】1,当需要的数为零时不仅要判断0出现的次数还要把没有零的自己算进去times++;}}return times;}
};