2086: [Poi2010]Blocks
Time Limit: 20 Sec Memory Limit: 259 MBSubmit: 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
1 2 1 1 5
1 2 3 4 5 6
Sample Output
HINT
Source
题解:乱搞
刚开始以为是单调栈,但是实际是单调的栈。
先分析可行的连续子串的特性,如果都能大于等于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");
}