一开始想到的是暴力,从1到n每次试一遍,但结果显然是超时的,时间复杂度大约是O(n2)。这里就不贴错误的代码了。
讲一下正确的思路是变环为链 从k到n 再从1到k 其实是一个环的过程
示例 -3 5 1 2 变为 -3 5 1 2 -3 5 1 就把一个环改成一个链 方便计算
每次 从 k开始 到 k+n-1 例如 k=1 到 1+4-1 k=2 到 2+4-1
所以题目变成了 k到k+n-1 内 任何一点的和都是非负的
如果再去暴力 应该还是会超时 用前缀和的思路比较好
s[i]=s[i-1]+a[i];
当你有了前缀和数组 ,再用单调队列 维护这个队列 使其长度不大于n 且保证队列的 头 是队列中最小的元素
之后 每次拿队列中最小的元素与i-n 来比较 如果队列中最下的元素都比i-n所代表的元素大 即认为没有问题
代码如下
小白写博客 有不理解或者错误请联系QQ2250362370
#include<iostream>
#include<deque>
using namespace std;
const int N = 2*1e6 + 10;
int arr[N],qsum[N];
deque<int>brr;
int main()
{
int n; cin >> n;
for (int i = 1; i <= n; i++)
{
cin >> arr[i];
}
for (int i = n + 1, k = 1; i <= n + n - 1; i++, k++)
{
arr[i] = arr[k];
}
/*for (int i = 1; i <= n + n - 1; i++)
{
cout << arr[i];
}*/
for (int i = 1; i <= n + n - 1; i++)
{
qsum[i] = qsum[i - 1] + arr[i];
}
int ans = 0;
for (int i = 1; i <= n + n - 1; i++)
{
while (!brr.empty() && qsum[brr.back()] > qsum[i])
{
brr.pop_back();
}
brr.push_back(i);
if (i >= n)
{
while (!brr.empty() && brr.front() <= i - n)
{
brr.pop_front();
}
if (qsum[brr.front()] - qsum[i - n] >= 0)
{
ans++;
}
}
}
cout << ans << endl;
return 0;
}