16和至少为 K 的最短子数组
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
可参考以下main函数:
int main()
{
vector<int> A;
int n,data,k;
cin>>n;
for(int i=0; i<n; i++)
{
cin>>data;
A.push_back(data);
}
cin>>k;
int result=Solution().shortestSubarray(A,k);
cout<<result<<endl;
return 0;
}
2.输入说明
首先输入A数组元素的数目n
然后输入n个整数
最后输入k
3.输出说明
输出一个整数,表示结果
4.范例
输入
3
2 -1 2
3
输出
3
5.代码
#include<iostream>
#include<stack>
#include<vector>
#include<math.h>
#include<algorithm>
using namespace std;
int shortestSubarray(vector<int>A, int k)
{
//部分用例会超时,复杂度O(n^2)
/*
//思想:1.计算每个位置的前缀和 2.依次遍历每个位置,计算符合连续和大于等于K的最小长度
int len = A.size()+1;
//范围:1 <= A.length <= 50000
int pre[50001];
//1.计算每个位置的前缀和,放在数组pre[]中
pre[0] = 0;
for (int i = 1; i <= A.size(); i++)
{
pre[i] = A[i-1] + pre[i-1];
//cout << "pre[" << i << "]=" << pre[i]<<endl;
}
//2.遍历寻找最短len
for (int i = 0; i < A.size(); i++)
{
for (int j = i + 1; j < A.size() +1; j++)
{
if (pre[j] - pre[i] >= k)
{
len = min(len, j - i);
}
}
}
return len>A.size()?-1:len;
*/
//官方题解
int n = A.size();
vector<long long> p(n + 1);
//cout << p[0] << endl; 结果: p[0]=0
for (int i = 1; i <= n; ++i) {
p[i] = p[i - 1] + A[i - 1]; //A[i-1]注意这里的下标
}
deque<int> q;//设置双端队列
q.push_back(0);//入队列
int ans = INT_MAX;//返回最短的子集长度
for (int i = 1; i <= n; ++i) {
while (q.size() && p[i] - p[q.front()] >= k) { //寻找满足连续和不小于K的最小子数组长度
//注意,一旦符合条件,就更新ans,并将该元素从头部删除(不用往后判断了)
ans = min(ans, i - q.front());
q.pop_front();
}
while (q.size() && p[q.back()] > p[i]) { //1.1将所有比p[i]大的元素都弹出 ,从尾部弹出
q.pop_back();
}
q.push_back(i);//1.2重新插入队列,保证队列是单调递增的状态
}
if (ans == INT_MAX) return -1;
return ans;
}
int main()
{
vector<int> A;
int n, data, k;
cin >> n;
for (int i = 0; i < n; i++)
{
cin >> data;
A.push_back(data);
}
cin >> k;
int result = shortestSubarray(A, k);
cout << result << endl;
return 0;
}