前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >LeetCode 862. 和至少为 K 的最短子数组(前缀和+deque单调栈)

LeetCode 862. 和至少为 K 的最短子数组(前缀和+deque单调栈)

作者头像
Michael阿明
发布2020-07-13 15:26:09
4420
发布2020-07-13 15:26:09
举报
文章被收录于专栏:Michael阿明学习之路

1. 题目

返回 A 的最短非空连续子数组的长度,该子数组的和至少为 K

如果没有和至少为 K 的非空子数组,返回 -1 。

代码语言:javascript
复制
示例 1:
输入:A = [1], K = 1
输出:1

示例 2:
输入:A = [1,2], K = 4
输出:-1

示例 3:
输入:A = [2,-1,2], K = 3
输出:3
 
提示:
1 <= A.length <= 50000
-10 ^ 5 <= A[i] <= 10 ^ 5
1 <= K <= 10 ^ 9

来源:力扣(LeetCode) 链接:https://leetcode-cn.com/problems/shortest-subarray-with-sum-at-least-k 著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

2. 解题

类似题目:

LeetCode 560. 和为K的子数组(前缀和差分)

LeetCode 523. 连续的子数组和(求余 哈希)

LeetCode 974. 和可被 K 整除的子数组(哈希map)

  • 参考官方思路
  • deque存储前缀和的下标,队内前缀和需要严格单调递增
  • 跟队首的差值 >= k 时,记录最小长度,删除队首
代码语言:javascript
复制
class Solution {
public:
    int shortestSubarray(vector<int>& A, int K) {
    	vector<int> presum(A.size()+1,0);
    	int i, minLen = INT_MAX, sum = 0;
    	deque<int> q;//存储下标,按照队内presum升序
    	q.push_back(0);//边界条件,前缀和为0,下标为0
    	for(i = 0; i < A.size(); ++i)
    	{
    		presum[i+1] = presum[i] + A[i];//前缀和
    		while(!q.empty() && presum[q.back()] >= presum[i+1])
    			q.pop_back();
    		while(!q.empty() && (presum[i+1]-presum[q.front()] >= K))
    		{
    			minLen = min(minLen, i+1-q.front());
    			q.pop_front();
    		}
    		q.push_back(i+1);
    	}
    	if(minLen == INT_MAX)
    		return -1;
    	return minLen;
    }
};
本文参与 腾讯云自媒体同步曝光计划,分享自作者个人站点/博客。
原始发表:2020/04/21 ,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 作者个人站点/博客 前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体同步曝光计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
目录
  • 1. 题目
  • 2. 解题
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档