【bzoj2086】 [Poi2010]Blocks

Description

给出N个正整数a[1..N]，再给出一个正整数k，现在可以进行如下操作：每次选择一个大于k的正整数a[i]，将a[i]减去1，选择a[i-1]或a[i+1]中的一个加上1。经过一定次数的操作后，问最大能够选出多长的一个连续子序列，使得这个子序列的每个数都不小于k。
总共给出M次询问，每次询问给出的k不同，你需要分别回答。

Input

第一行两个正整数N (N <= 1,000,000)和M (M <= 50)。
第二行N个正整数，第i个正整数表示a[i] (a[i] <= 10^9)。
第三行M个正整数，第i个正整数表示第i次询问的k (k <= 10^9)。

Output

共一行，输出M个正整数，第i个数表示第i次询问的答案。

Sample Input

5 6
1 2 1 1 5
1 2 3 4 5 6

Sample Output

5 5 2 1 1 0

平均数大于K的区间一定可以，每次减K求前缀和，单调栈维护一下

#include<iostream>#include<cstdlib>#include<cstdio>using namespace std;const int N=1000005;int n,m,a[N],x,top,q[N],ans;long long sum[N];inline void solve(int x){    sum[0]=0;    for (int i=1;i<=n;++i)        sum[i]=sum[i-1]+a[i]-x;    for (int i=1;i<=n;++i)        if (sum[q[top]]>sum[i])q[++top]=i;    for (int i=n;i>=1;--i){        while (top && sum[i]>=sum[q[top-1]])top--;        ans=max(ans,i-q[top]);    }    printf ("%d",ans);}int main (){    scanf ("%d%d",&n,&m);    for (int i=1;i<=n;++i)        scanf ("%d",&a[i]);    for (int i=1;i<=m;++i){        scanf ("%d",&x);        ans=0;        solve(x);        if (i!=m)printf (" ");        else puts("");    }    return 0;}