题意:给你一个N,K。表示又N个人,每个人有一定的钱数,每天最富有的人要给最穷的人给一块钱,问k天后最富有的人(最大值)跟最穷的人(最小值)的差值为多少?
思路:因为时间越长(天数经过的越多),最大值就会越小,同理最小值就会越大,所以分别二分最富有的人的钱数(最大值),最穷的人的钱数(最小值),两个相减为答案,注意要用long long。
#include<bits/stdc++.h>
using namespace std;
const int MAXN = 5e5 + 5;
const int INF = 0x3f3f3f3f;
long long n,k,a[MAXN];
long long sum;
long long check1(int m)//最小值
{
long long sum = 0;
for(int i = 0;i < n; i++)
{
if(a[i] < m) sum = sum + (m - a[i]);//大于当前二分的最大值m的部分的和累加起来
}
return sum;
}
long long check2(int m)//最大值
{
long long sum = 0;
for(int i = 0;i < n; i++)
{
if(a[i] > m) sum = sum + (a[i] - m);//大于当前二分的最小值m的部分的和累加起来
}
return sum;
}
int main()
{
while(~scanf("%lld%lld",&n,&k))
{
sum = 0;
long long MAX = -1,MIN = INF;
for(int i = 0; i < n; i++)
{
scanf("%lld",&a[i]);
MAX = max(MAX,a[i]);MIN = min(MIN,a[i]);
sum += a[i];
}
//求最穷的人的钱数
long long poor,rich;
long long l = MIN,r = sum / n,m = l + (r - l) / 2;
while(l <= r)
{
if(check1(m) <= k)
{
poor = m;l = m + 1;
}
else if(check1(m) > k) r = m - 1;
m = l + (r - l) / 2;
}
//求最富的人的钱数
l = sum / n,r = MAX;
if(sum % n) l++;
m = l + (r - l) / 2;
while(l <= r)
{
if(check2(m) <= k)
{
rich = m;r = m - 1;
}
else if(check2(m) > k) l = m + 1;
m = l + (r - l) / 2;
}
printf("%lld\n",rich - poor);
}
return 0;
}
/*
4 1
1 1 4 2
3 1
2 2 2
5 1
1 1 1 1 2
*/