862. 和至少为 K 的最短子数组

给你一个整数数组 nums 和一个整数 k ,找出 nums 中和至少为 k 的 最短非空子数组 ,并返回该子数组的长度。如果不存在这样的 子数组 ,返回 -1 。子数组 是数组中 连续 的一部分。

示例 3:

  • 输入:nums = [2,-1,2], k = 3
  • 输出:3

题目链接

暴力

首先将前缀和算出来。然后通过前缀和去计算区间和。记录下最小的满足题意的区间长度即可

class Solution {
public:
    int shortestSubarray(vector<int>& nums, int k) {
        vector<int>sum(nums.size(),0);
        sum[0]=nums[0];
        int res=INT_MAX;
        //前缀和
        for(int i=1;i<nums.size();i++){
            sum[i]=sum[i-1]+nums[i];
        }
        for(int i=0;i<nums.size();i++){
            if(sum[i]>=k){
                res=min(res,i+1);
                //printf("sum[%d]\n",i);
                break;
            }
        }
        for(int idx=1;idx<nums.size();idx++){
            for(int j=idx;j<nums.size();j++){
                if(sum[j]-sum[idx-1]>=k){
                    res=min(res,j-idx+1);
                    //printf("sum[%d~%d]\n",idx,j);
                    break;
                }
            }
        }
        return res==INT_MAX?-1:res;
    }
};

很显然暴力是必然会TEL的

前缀和加队列

当然是学习了灵佬的题解才做出来的啦。孱弱垃圾博主怎么可能单吃困难呢

题解地址

找到最小j-i满足 j>i && presum[j+1]-presum[i]>=k

优化一

当presum[j+1]-presum[i]>=k已经成立的时候。需要将i出队。因为后面的数已经比j大了,区间长度不可能小于j-i+1

优化二

当presum[j]<=presum[i]的时候。这时,若后面存在x满足x-presum[i]>=k。那也必然满足x-presum[j]>=k。应该将i出队

class Solution {
public:
    int shortestSubarray(vector<int>& nums, int k) {
        int n=nums.size();
        vector<long>sum(n+1);
        int res=n+1;
        for(int i=0;i<n;i++){
            sum[i+1]=sum[i]+nums[i];
        }
        deque<int>p;
        for(int i=0;i<=nums.size();i++){
            long cur_num=sum[i];
            while(!p.empty()&&cur_num-sum[p.front()]>=k){
                res=min(res,i-p.front());
                p.pop_front();
            }
            while(!p.empty()&&cur_num<=sum[p.back()]){
                p.pop_back();
            }
            p.push_back(i);
        }
        return res>nums.size()?-1:res;
    }
};

原文地址:http://www.cnblogs.com/SkyDusty/p/16828621.html

1. 本站所有资源来源于用户上传和网络,如有侵权请邮件联系站长! 2. 分享目的仅供大家学习和交流,请务用于商业用途! 3. 如果你也有好源码或者教程,可以到用户中心发布,分享有积分奖励和额外收入! 4. 本站提供的源码、模板、插件等等其他资源,都不包含技术服务请大家谅解! 5. 如有链接无法下载、失效或广告,请联系管理员处理! 6. 本站资源售价只是赞助,收取费用仅维持本站的日常运营所需! 7. 如遇到加密压缩包,默认解压密码为"gltf",如遇到无法解压的请联系管理员! 8. 因为资源和程序源码均为可复制品,所以不支持任何理由的退款兑现,请斟酌后支付下载 声明:如果标题没有注明"已测试"或者"测试可用"等字样的资源源码均未经过站长测试.特别注意没有标注的源码不保证任何可用性