1. 题目
返回 A 的最短的非空连续子数组的长度,该子数组的和至少为 K 。
如果没有和至少为 K 的非空子数组,返回 -1 。
示例 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 时,记录最小长度,删除队首
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,下标为0for(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;}
};