codeforces672Robin Hood（经典二分）

怎么都没想到是二分，我真弱爆了。

二分求出k天后最小的人的财富，

理由：

最小的人的财富必然小于或等于平均数向下取整（记为la），

所以范围在0~la之间，二分答案。

再用二分求出k天后最大的人的财富。相减即得答案。

注意：sum等可能会爆int.

#include <iostream>
#include <algorithm>
#include <string.h>
#include <stdio.h>
using namespace std;
#define ms(X) memset(X,0,sizeof(X))
#define Inf 0x3fffffff
typedef long long ll;
int arr[502000];
int main(int argc, char const *argv[])
{
int n,k;
ll sum=0;
cin>>n>>k;
for(int i=0;i<n;i++)
{scanf("%d",arr+i);
sum+=(ll)arr[i];
}
int l=0,r=(int)(sum/n)+1,ans1=0,mid;
while(l<r)
{
mid=(l+r)>>1;
ll tmp=0;
for(int i=0;i<n;i++)
if(mid>arr[i]) tmp+=mid-arr[i];
if(tmp<=k) ans1=mid,l=mid+1;
else r=mid;
}
l=(int)((sum+n-1)/n),r=Inf;
int ans2=0;
while(l<r)
{
mid=((r-l)>>1)+l;
ll tmp=0;
for(int i=0;i<n;i++)
if(mid<arr[i]) tmp+=arr[i]-mid;
if(tmp<=k) ans2=mid,r=mid;
else l=mid+1;
}
printf("%d\n",ans2-ans1 );
return 0;
}