BZOJ 2600: [Ioi2011]ricehub|暴力|中位数

Posted 2023-02-04 ws_yzy

答案显然在一个区间里

从左到右枚举右端点，更新左端点直到cost<=B

米仓显然是中点（中位数）

每次修改中点 维护一下cost就好了

复杂度O（n）

#include<cstdio>
#include<cstdlib>
#include<cmath>
#include<cstring>
#include<queue>
#include<vector>
#include<set>
#include<map>
#include<iostream>
#include<algorithm>
#define T 111111
#define ll long long
using namespace std;
ll sc()

	ll i=0;char c=getchar();
	while(c>'9'||c<'0')c=getchar();
	while(c>='0'&&c<='9')i=i*10+c-'0',c=getchar();
	return i;

ll a[T],B,cost,mid;
int n,L,l=1,r=1,ans=1;
int main()

	n=sc(),L=sc(),B=sc();
	for(int i=1;i<=n;i++)a[i]=sc();
	mid=1;cost=0;
	for(int i=2;i<=n;i++)
	
		ll md=i+l>>1;
		ll change=a[i]-a[md];
		if(md!=mid) change+=(a[md]-a[mid])*(mid-l)-(i-md-1)*(a[md]-a[mid]);
		cost+=change;
		mid=md;
		if(cost<=B)
			ans=max(i-l+1,ans);
		else
		
			while(cost>B)
			
				l++;
				md=(l+i)>>1;
				change=a[l-1]-a[mid];
				if(md!=mid) change+=(a[md]-a[mid])*(mid-l-(i-md));
				cost+=change;
				mid=md;
			
			ans=max(ans,i-l+1);
		
	
	cout << ans;
	return 0;