CodeForces 671 B.Robin Hood（水~）

Description
一个长度为n的序列，每次可以把其中一个值减一，一个值加一，求k次操作后极差最小值
Input
第一行两个整数n,k表示序列长度和操作数，之后n个整数a[i]表示该序列(1<=n<=5e5,0<=k<=1e9,1<=a[i]<=1e9)
Output
输出k次操作后极差最小值
Sample Input
4 1
1 1 4 2
Sample Output
2
Solution
给a[i]排个序，同样的a[i]值合并，得到若干层a[i]，先用k把小的值一层层往大提到l，再把大的值一层层往小压到r，如果l>=r说明k足够把最大值和最小值充分接近，令sum为a[i]的和，如果sum%n=0说明k可以把所有数变成平均数，那么答案就是0，否则只能让最大值比最小值大1，答案是1；如果l < r说明答案就是r-l
Code

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<vector>
#include<queue>
#include<map>
#include<set>
#include<ctime>
using namespace std;
typedef long long ll;
#define INF 0x3f3f3f3f
#define maxn 555555
struct node
{
    int c,num;
    bool operator<(const node&b)const
    {
        return c<b.c;
    }
}a[maxn];
int k,n;
ll sum[maxn];
int main()
{
    while(~scanf("%d%d",&n,&k))
    {
        ll tol=0;
        for(int i=0;i<n;i++)scanf("%d",&a[i].c),tol+=a[i].c;
        sort(a,a+n);
        int res=0;
        for(int i=0;i<n;i++)
        {
            int temp=1;
            while(i<n-1&&a[i].c==a[i+1].c)i++,temp++;
            a[res].c=a[i].c,a[res++].num=temp;
        }
        //for(int i=0;i<res;i++)printf("%d %d\n",a[i].c,a[i].num);
        int l=-1,r=-1,num;
        sum[0]=0,num=a[0].num;
        for(int i=1;i<res;i++)
        {
            sum[i]=sum[i-1]+1ll*num*(a[i].c-a[i-1].c);
            if(sum[i]>k)
            {
                l=a[i-1].c+(k-sum[i-1])/num;
                break;
            }
            num+=a[i].num; 
        }
        sum[res-1]=0,num=a[res-1].num;
        for(int i=res-2;i>=0;i--)
        {
            sum[i]=sum[i+1]+1ll*num*(a[i+1].c-a[i].c);
            if(sum[i]>k)
            {
                r=a[i+1].c-(k-sum[i+1])/num;
                break;
            }
            num+=a[i].num; 
        }
        //printf("l=%d,r=%d\n",l,r);
        if(l>=r)printf("%d\n",tol%n==0?0:1);
        else printf("%d\n",r-l);
    }
    return 0;
}