Codeforces 672D Robin Hood 二分判定+思维

点击打开链接

//题意:有n个数 k个操作 每次把最大值+1,最小值-1 问k次后的最大值和最小值之差,每个数都相等时停止
//n<=5e5 k<=1e9 二分最大值x和最小值y后判定
// 。二分一个符合情况的最大值max。那么就要把比max大的数补给比max小的数。
//假设给出部分为A 最能得到的最大部分为B，如果A<=k&&B>=A A<=k mx太大  A<=B是所给出的不能大于所能得到的最大 

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int inf=1e9;
const int N=5e5+20;
int n,k;
int a[N];
bool mx(int x)
{
	ll A=0,B=0;
	for(int i=1;i<=n;i++)
	{
		if(a[i]>x)
		A+=a[i]-x;
		else
		B+=x-a[i];//比max小的最多能得到多少 
	}
	//<k说明x太大 
	if(A<=k&&A<=B)//给出的部分A必须小于天数&&(给出的)A<=B(能得到的最大值) 
	return 1;
	else
	return 0;
}
bool mn(int x)
{
	ll A=0,B=0;
	for(int i=1;i<=n;i++)
	{
		if(a[i]<x)
		B+=x-a[i];
		else
		A+=a[i]-x;
	}
	//B<k x太小 
	if(B<=k&&B<=A)//得到的B小于天数 
	return 1;
	else
	return 0;
}
int calc_mx()
{
	int l=1,r=inf,Max;
	while(l<=r)
	{
		int mid=(l+r)/2;
		if(mx(mid))//
		{
			Max=mid;
			r=mid-1;
		}
		else
		l=mid+1;
	}
	return Max;
}
int calc_mn()
{
	int l=1,r=inf,Min;
	while(l<=r)
	{
		int mid=(l+r)/2;
		if(mn(mid))//min太小 
		{
			Min=mid;
			l=mid+1;//
		}
		else
		r=mid-1; 
	}
	return Min;
}
int main()
{
	while(cin>>n>>k)
	{
		for(int i=1;i<=n;i++)
		scanf("%d",&a[i]);
		
		int Max=calc_mx();
		int Min=calc_mn();
		//cout<<Max<<' '<<Min<<endl;
		cout<<Max-Min<<endl;
	}

	return 0;	
}