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;
}