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+;
 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=;i<m;cnt[i++]=n/m)
         s.insert(i);
     for(int i=;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=;i<=n;i++)
         cout<<a[i]<<" ";
     cout<<endl;
     return ;
 }