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不同，你需要分别回答。

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

Solution

---

nmlogn的做法使比较好想的。我们把每个数字减去k，等价于要找最长一段和为非负整数的子区间。考虑枚举右端点，我们把（s[i]，i）看作二维平面上的点然后线段树就行了，其中s是前缀和。据说卡一卡能过？！

考虑开一个栈，每次当s[i]小于栈顶时就入栈。我们从n到1倒着做，弹栈直到栈顶大于等于s[r]，用这些栈顶更新答案
这样为什么是对的呢？首先这样的答案一定是合法的，并且我们漏掉的区间一定不会比目前答案更优秀

Code

---

#include <stdio.h>
#include <string.h>
#include <algorithm>
#define rep(i,st,ed) for (int i=st;i<=ed;++i)
#define drp(i,st,ed) for (int i=st;i>=ed;--i)
 
typedef long long LL;
const int N=2000005;
 
int a[N],stack[N];
 
LL s[N];
 
int read() {
    int x=0,v=1; char ch=getchar();
    for (;ch<'0'||ch>'9';v=(ch=='-')?(-1):(v),ch=getchar());
    for (;ch<='9'&&ch>='0';x=x*10+ch-'0',ch=getchar());
    return x*v;
}
 
int main(void) {
    int n=read(),m=read();
    rep(i,1,n) a[i]=read();
    for (;m--;) {
        int k=read(),ans=0;
        int top=1;
        rep(i,1,n) {
            s[i]=s[i-1]+a[i]-k;
            if (s[i]<s[stack[top]]) stack[++top]=i;
        }
        drp(i,n,1) {
            while (top&&s[stack[top]]<=s[i]) {
                ans=std:: max(ans,i-stack[top]);
                top--;
            }
        }
        printf("%d\n", ans);
    }
    return 0;
}