力扣Hot100-T10和为k的子数组(前缀和+哈希表)注意思路

中等

给你一个整数数组 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,当前数之和也为-1

            if(sum1.find(need)!=sum1.end()){
                if(need==sum[i]){
                    times+=(sum1[need]-1); //[-1,-1,1]中重复计算了
               
                }
                
                else
                 times+=sum1[need];

            }
             if (need==0){//注意【0,1】1,当需要的数为零时不仅要判断0出现的次数还要把没有零的自己算进去
             
                times++;
            }



        }
       
        
        return times;
       
       




    }
};

  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

半截詩

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值