***Leetcode 862. Shortest Subarray with Sum at Least K | 区间和大于某个数

最新推荐文章于 2025-02-03 15:37:10 发布

Z-Pilgrim

最新推荐文章于 2025-02-03 15:37:10 发布

阅读量1.1k

点赞数

CC 4.0 BY-SA版权

分类专栏： LeetCode题解

本文链接：https://blog.youkuaiyun.com/u011026968/article/details/80887485

LeetCode题解专栏收录该内容

271 篇文章

订阅专栏

本文介绍了一种使用优先队列解决LeetCode上最短子数组问题的方法。通过维护一个优先队列来跟踪满足条件的子数组，实现快速查找最短长度。此算法适用于数组中元素求和至少为k的情况。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

https://leetcode.com/problems/shortest-subarray-with-sum-at-least-k/description/

堆

思路也比较简单

i1, i2, i3,i4,i5

如果i2->i3这个区间符合条件，那么如果存在另一个区间符合>=k, 这个区间右端点>i3，则左端点一定也大于i2

class Solution {
public:
	int shortestSubarray(vector<int>& A, int k) {
		priority_queue< pair<int, int>, vector< pair<int, int> >, greater< pair<int, int> > > pq;
		int cur = 0, ans = A.size() + 1;
		for (int i = 0; i < A.size(); i++) {
			cur += A[i];
			if (cur >= k) {
				ans = min(ans, i + 1);
			}
			while (pq.size() && cur - pq.top().first >= k) {
				ans = min(ans, i - pq.top().second );
                pq.pop();
			}
			pq.push( make_pair(cur, i) );
		}
		return ans == A.size()+1? -1 : ans;
	}
};