传送门:
D. Max GEQ Sum
题意:给你一个序列,问是否所有区间都满足 区间最大值>=区间和。
分析:对于每个a[i],都往左右扩到第一个大于a[i]的位置记为l[i],r[i],那么在[l[i],r[i]]内a[i]是区间的最大值。此时如果在[l[i],r[i]]内最大的区间和都<=a[i],那么这个a[i]满足条件。所有a[i]都满足条件的话YES。实现的话这里用了单调队列+st表。
代码:
#include<bits/stdc++.h>
using namespace std;
#define int long long
const int maxlog=21;
const int maxn=2e5+5;
int table1[maxn][maxlog],table2[maxn][maxlog];
int a[maxn],pre[maxn];
int l[maxn],r[maxn];
int askmax(int l,int r)
{
int st=log2(r-l+1);
return max(table1[l][st],table1[r-(1<<st)+1][st]);
}
int askmin(int l,int r)
{
if(!r) return 0;//处理边界为0的情况
int ans=1e18;//
if(!l) ans=0,l=1;//
int st=log2(r-l+1);
return min(ans,min(table2[l][st],table2[r-(1<<st)+1][st]));
}
signed main()
{
int t;
cin>>t;
while(t--)
{
int n;
cin>>n;
for(int i=1;i<=n;i++)
{
cin>>a[i];
pre[i]=pre[i-1]+a[i];
table2[i][0]=table1[i][0]=pre[i];
}
for(int st=1;(1<<st)<=n;st++)
{
for(int i=1;i+(1<<st)-1<=n;i++)
{
table1[i][st]=max(table1[i][st-1],table1[i+(1<<(st-1))][st-1]);
table2[i][st]=min(table2[i][st-1],table2[i+(1<<(st-1))][st-1]);
}
}
stack<int>s1,s2;
for(int i=1;i<=n;i++)
{
while(!s1.empty()&&a[s1.top()]<=a[i]) s1.pop();
l[i]=s1.empty()?0:s1.top();
s1.push(i);
}
for(int i=n;i>=1;i--)
{
while(!s2.empty()&&a[s2.top()]<=a[i]) s2.pop();
r[i]=s2.empty()?0:s2.top();
s2.push(i);
}
for(int i=1;i<=n;i++)
{
if(!l[i]) l[i]=1;
else l[i]++;
if(!r[i]) r[i]=n;
else r[i]--;
}
int f=1;
for(int i=1;i<=n;i++)
{
int mn=askmin(l[i]-1,i-1);//因为是前缀和所以要-1
int mx=askmax(i,r[i]);
if(a[i]<mx-mn) f=0;
}
if(f) puts("YES");
else puts("NO");
}
}