【CodeForces】671 B. Robin Hood

【题目】B. Robin Hood

【题意】给定n个数字的序列和k次操作，每次将序列中最大的数-1，然后将序列中最小的数+1，求最终序列极差。n<=5*10^5,0<=k<=10^9,1<=ai<=10^9。

【算法】模拟

【题解】关键在于，增加和减少可以分开操作。

将数列排序，从小到大增加前面若干个数到同一个数直到k次，再从大到小减少后面若干个数到同一个数直到k次，就可以得到结果。

也可以不排序直接二分”同一个数”找到恰好k次的位置。

#include<cstdio>
#include<algorithm>
int n,k,a[];
void solve(){
    std::sort(a+,a+n+);
    int t=,b=,s=k;
    while(t<n&&1ll*t*(a[t+]-b)<=s)s-=1ll*t*(a[t+]-b),b=a[++t];
    b=s/t;s%=t;
    for(int i=;i<=t;i++)a[i]=a[t]+b+(i>t-s);
    for(int i=;i<=n;i++)a[i]=-a[i];
}
int main(){
    scanf("%d%d",&n,&k);
    for(int i=;i<=n;i++)scanf("%d",&a[i]);
    solve();solve();
    printf("%d",a[]-a[n]);
    return ;
}