CodeForces - 999D——Equalize the Remainders

題目鏈接：https://vjudge.net/contest/336720#problem/J

題意：給出一個序列A一共n個數，給出一個m是n的因子。每次可以進行一個操作，使得A【i】+1。求至少進行多少次操作可以滿足序列A中每個元素%m後的序列恰好有n/m個【0~m-1】.

#include<stdio.h>
#include<algorithm>
#include<string.h>
#include<set>
#define ll long long
using namespace std;
ll n,m;
ll A[200010],re[200010];
set<ll> se;
int main()
{
    scanf("%lld%lld",&n,&m);
    for(ll i=1; i<=n; i++)
        scanf("%lld",&A[i]);
    for(ll i=0; i<m; i++)
        se.insert(i);
    ll ans=0;
    for(ll i=1; i<=n; i++)
    {
        int d=A[i]%m,z;
        if(d>*se.rbegin()) z=*se.begin();
        else z=*se.lower_bound(d);
        ans+=((z)-d+m)%m;
        A[i]+=((z)-d+m)%m;
        re[z]++;
        if(re[z]==(n/m))
            se.erase(z);
    }
    printf("%lld\n",ans);
    for(ll i=1; i<=n; i++) printf("%lld%s",A[i],i==n?"\n":" ");
}