分班级
题目链接:分班级
题目描述:
Zoro是一个有强迫症的人,他喜欢均衡。
17级的新生要开学了,起初所有班级是按照学生的来源地分的,各班人数非常不合理。于是老师要求Zoro来让各班人数均衡一下。
由于学校系统陈旧,每次调换只能是Zoro自己手动的把一个学生从一个班级拉到另外一个班级,由于Zoro有强迫症,他每次会找出班级人数最多的那个拉出一个人转到一个人数最少的班级,而且每次操作,耗费一次权限,而他的账号只有k次权限。老师最后会看所有班级人数差的最大值。
由于班级实在太多,Zoro要做完需要很长时间,你能不能帮助Zoro先计算出他工作完成后所有班级人数差的最大值报告给老师。
输入:
第一行输入两个整数n和k,分别代表班级数量和Zoro账号的操作权限次数。 接下来一行n个整数 第i个数字代表第i个班级有ci个人。
(1<=n<=500000,0<=k<=1e9,0<= ci <=1e9)
输出:
输出一个整数表示最后所有班级人数差的最大值。
样例输入:
5 1
1 2 3 4 5
样例输出:
2
思路:我们先假设k无限大,
这时k次操作后的差值最大为1,即最大值为sum/n+1,最小值为sun/n(sum%n=0时差值为0)
但是如果k的大小不足以使最大值和最小值达到这种状态该怎么办?
这时的最大值肯定在[sum/n+1,maxx]范围内,最小值在[minx,sum/n]范围内
因此我们可以直接二分枚举最终的最大值和最小值,判断是否满足≤k天即可
代码:
#include<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;
typedef long long LL;
const LL inf=0x3f3f3f3f3f3f3f3f;
const int maxn=5e5+10;
LL a[maxn];
int n,k;
bool check_min(LL x)//以x为最小值,判断是否满足≤k天
{
LL tot=0;
for(int i=1; i<=n; ++i)
if(a[i]<x)
tot+=x-a[i];
return tot<=k;
}
bool check_max(LL x)//以x为最大值,判断是否满足≤k天
{
LL tot=0;
for(int i=1; i<=n; ++i)
if(a[i]>x)
tot+=a[i]-x;
return tot<=k;
}
int main()
{
scanf("%d%d",&n,&k);
LL minx=inf,maxx=0,sum=0;
for(int i=1; i<=n; ++i)
{
scanf("%lld",&a[i]);
maxx=max(maxx,a[i]);
minx=min(minx,a[i]);
sum+=a[i];
}
LL lower=sum/n,upper=lower;
if(sum%n)
upper+=1;
LL le=minx,ri=lower,mid;
while(le<=ri)//枚举最小值
{
mid=le+ri>>1;
if(check_min(mid))
{
minx=mid;
le=mid+1;
}
else
ri=mid-1;
}
le=upper,ri=maxx;
while(le<=ri)//枚举最大值
{
mid=le+ri>>1;
if(check_max(mid))
{
maxx=mid;
ri=mid-1;
}
else
le=mid+1;
}
printf("%lld\n",maxx-minx);
return 0;
}