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]
,则此区间满足:
即:
于是维护一个单调递减的单调栈即可。复杂度 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
的规模的规模可达
当然有的大犇会说这不是显然要弹栈吗。但是我不是大犇,至少现在不是。
#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;
}