bzoj 2086: [Poi2010]Blocks （乱搞）

2086: [Poi2010]Blocks

Time Limit: 20 Sec   Memory Limit: 259 MB
Submit: 469   Solved: 213
[ Submit][ Status][ Discuss]

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

HINT

Source

[ Submit][ Status][ Discuss]

题解：乱搞

刚开始以为是单调栈，但是实际是单调的栈。

先分析可行的连续子串的特性，如果都能大于等于k，那么这一子串的可以平均值一定也大于等于k，因为这种移动方式只要连续可以给子串中的任意位置增加答案。

那么怎么维护最长的呢？我们考虑将所有的数减去k，然后求前缀和，那么如果后面的某一个位置的sum值大于前面的某一位置，那么这段区间一定是合法的。然后我们维护一个单调的栈，就是只有sum[i]<sum[st[top]]的时候才将当前位置入栈，注意是不弹栈的。哪这样舍弃的状态一定没用了吗？舍弃的位置可能产生合法的区间但是一定不是最长的，因为前面的位置既然比他小，那么更容易满足比后面的某个位置大，而且位置还靠前。

我们得到单调的栈之后，从后往前计算答案，如果栈顶元素小于sum[i]那么栈顶元素出栈，直到不满足条件或栈为空。那么这样会错失合法的区间，但是随着右端点的左移，左端点只有左移才能得到更优的答案，所以直接跳过的一定不是最优答案。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define N 1000003
#define LL long long 
using namespace std;
int n,m,st[N],top;
LL sum[N],a[N];
int main()
{
	freopen("a.in","r",stdin);
	freopen("my.out","w",stdout);
	scanf("%d%d",&n,&m);
	for (int i=1;i<=n;i++) scanf("%I64d",&a[i]);
	for (int i=1;i<=m;i++) {
		LL k; scanf("%I64d",&k); int ans=0;
		for (int j=1;j<=n;j++) {
		 sum[j]=sum[j-1]+a[j]-k;
		 if (a[j]>=k) ans=1;
	    }
		top=0; st[++top]=0;
		for (int j=1;j<=n;j++) {
			if (!top) st[++top]=j;
			else if (sum[st[top]]>sum[j]) st[++top]=j;
		}
		for (int j=n;j>=1;j--) {
			while (st[top]>j) top--;
			while (sum[st[top]]<=sum[j]&&top) top--;
		    ans=max(ans,j-st[top+1]);
		}
		if(i!=m)  printf("%d ",ans);
		else printf("%d\n",ans);
	}
	//printf("\n");
}