BZOJ2086[Poi2010] Blocks

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

Solution：

最近做题好像又有一点急躁了，该慢一点了，巩固自己学过的东西，才会不至于陷入其他大犇都会写而你这个蒟蒻不会写的困境。

比较明显的考虑终态枚举。假设最后的区间为$[L,R]$，则此区间满足：

$$sum[R]-sum[L-1]>=k*(R-L+1)$$
即：
$$sum[R]-kR-k>=sum[L-1]-kL$$
于是维护一个单调递减的单调栈即可。复杂度 $O(nmlogn)$。

struct Node{int id;ll x;}stk[M];
int A[M],top,n;
ll sum[M];
int find(ll x){
    int L=1,R=top,res=-1;
    while(L<=R){
        int mid=(L+R)>>1;
        if(stk[mid].x<=x){
            res=mid;
            R=mid-1;
        }else L=mid+1;
    }
    return res;
}
void solve(int k){
    top=0;
    int ans=0;
    for(int R=1;R<=n;R++){
        ll rs=sum[R-1]-1LL*k*R;
        if(!top||rs<stk[top].x)stk[++top]=(Node){R,rs};
        int t=find(sum[R]-1LL*k*R-k);
        int l=R-stk[t].id+1;
        if(t!=-1&&ans<l)ans=l;
    }
    printf("%d",ans);
}

但这题$n$的规模的规模可达$10^6$，这样并不能通过。于是我们观察这个单调栈：值单调递减，下标单调递增，因为我们要下标间隔最大，于是从后往前扫描$R$，不满足的直接弹栈，因为接下来如果取的值在弹栈了的元素中的话的肯定不优。于是就解决了这道题，复杂度$O(nm)$。

---

当然有的大犇会说这不是显然要弹栈吗。但是我不是大犇，至少现在不是。

#include<stdio.h>
#include<iostream>
#define ll long long
#define M 1000005
#define INF (1LL<<61)
using namespace std;
struct Node{int id;ll x;}stk[M];
int A[M],top,n;
ll sum[M];
inline void Rd(int &res){
    char c;res=0;
    while(c=getchar(),!isdigit(c));
    do{
        res=(res<<1)+(res<<3)+(c^48);
    }while(c=getchar(),isdigit(c));
}
void solve(int k){
    top=0;
    int ans=0;
    for(int i=1;i<=n;i++){
        ll rs=sum[i-1]-1LL*k*i;
        if(!top||rs<stk[top].x)stk[++top]=(Node){i,rs};
    }
    int max=top;
    for(int R=n;R>=1;R--){
        if(!top)break;
        ll rs=sum[R]-1LL*k*R-k;
        while(top&&rs>=stk[top].x)top--;
        int t=R-stk[top+1].id+1;
        if(top<max&&t>ans)ans=t;
    }
    printf("%d",ans);
}
int main(){
    int m;
    Rd(n);Rd(m);
    for(int i=1;i<=n;i++){
        Rd(A[i]);
        sum[i]=sum[i-1]+A[i];
    }
    for(int i=1;i<=m;i++){
        int k;
        Rd(k);
        if(i!=1)putchar(' ');
        solve(k);
    }
    return 0;
}