题目链接
https://www.lydsy.com/JudgeOnline/problem.php?id=2086
思路
这就有点妙了
题目意思就是让你求平均数>=k的最长序列
先求出a[i]-k的前缀和
就是求sum[i]-sum[j]>=0的最大i-j
当(j<=k<=i sum[j]<=sum[k])
更新i的时候,k就不如j优
所以处理出来一个单调上升的数组(stak),那答案就在里面选啦
倒序更新,单调栈一直减减就好
因为如果用stak[top]更新了i,因为i是倒序枚举,递减的,stak又是上升的
所以只有top--才能更新答案
妙啊
错误
边界好麻烦
代码
#include <bits/stdc++.h>
#define ll long long
using namespace std;
const ll N=1e6+7;
ll read() {
ll x=0,f=1;char s=getchar();
for(;s>'9'||s<'0';s=getchar()) if(s=='-') f=-1;
for(;s>='0'&&s<='9';s=getchar()) x=x*10+s-'0';
return x*f;
}
ll n,m,k,a[N],stak[N],top,sum[N];
void solve() {
for(ll i=1;i<=n;++i) sum[i]=sum[i-1]+a[i]-k;
stak[top=1]=0;
for(ll i=1;i<=n;++i)
if(sum[stak[top]]>sum[i]) stak[++top]=i;
// for(ll i=1;i<=top;++i) cout<<sum[stak[i]]<<"! ";puts("");
ll ans=0;
for(ll i=n;i>=1;--i) {
while(top>1&&sum[i]>=sum[stak[top-1]]) top--;
// if(top) top++;
// cout<<sum[i]<<" "<<sum[stak[top]]<<" !
";
// if(sum[i]>=sum[stak[top]]&&top)
// cout<<stak[top]<<" "<<i<<" "<<top<<"
";
if(sum[stak[top]]<=sum[i])
ans=max(ans,i-stak[top]);
}
printf("%lld ",ans);
}
int main() {
// freopen("3.in","r",stdin);
n=read(),m=read();
for(ll i=1;i<=n;++i) a[i]=read();
for(ll i=1;i<=m;++i) {
k=read();
solve();
}
return 0;
}