CodeForces-999D Equalize the Remainders (贪心+神奇的STL)

题意：给你一个n,m;其中n一定能被m整除，然后给你n个数

有一种操作   选择n个数中的任意一个，使其+1；

条件：

　　Ci 属于[0,m-1]  C_i代表a_i模m的余数为i的个数 且都等于n/m;

　　(

　　　　比如n=4,m=2;n/m=2;

　　　　a1=0,a2=1,a3=2,a4=3;

　　　　C_i属于[0,1]；

　　　　a1%m=0;a2%m=1;a3%m=0;a4%m=1;

　　　　所以C₁=C₂=n/m=2;

　　)

求：最少需要操作几次才能满足以上要求

代码如下：

#include <iostream>
#include <cstdio> 
#include <set> 
#include <algorithm> 
using namespace std;
typedef long long ll;
const int maxn=2e5+5;
ll cnt[maxn],a[maxn],val,res,ans;
set<ll> s;
set<ll> ::iterator it;
int main()
{
    int n,m;
    cin>>n>>m;
    for(int i=0;i<m;cnt[i++]=n/m)
        s.insert(i);
    for(int i=1;i<=n;i++)
    {
        cin>>a[i];
        val=a[i]%m;
        if(val>(*s.rbegin()))res=*s.begin();
        else res=*s.lower_bound(val);
        if(!--cnt[res])s.erase(res);
        ans=ans+(res-val+m)%m;
        a[i]=a[i]+(res-val+m)%m;
    }
    cout<<ans<<endl;
    for(int i=1;i<=n;i++)
        cout<<a[i]<<" ";
    cout<<endl;
    return 0;
}