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. 因为资源和程序源码均为可复制品,所以不支持任何理由的退款兑现,请斟酌后支付下载
声明:如果标题没有注明"已测试"或者"测试可用"等字样的资源源码均未经过站长测试.特别注意没有标注的源码不保证任何可用性