codeforces 999D Equalize the Remainders

题目：点击打开链接

题意：给你一个含有n个整数的数组a1,a2,…,an，和一个正整数m。保证m是n的因数。 在单次移动中，你可以选择在1到n之间的任一位置的数ai加1. 计算cr（0~m-1）——每个元素除以m之后的余数r。换句话说，对于每个余数， 找到与它相对应的元素。 你的任务是改变数组的元素使得c0=c1=…=cm-1=n/m; 找到满足上述要求的最小的需要改变的次数 。

分析：贪心，首先记录每个取余m后的结果数的个数，如果个数不大于n/m则不用改变，否则（需要改变）插入set，然后二分查找出大于该数的且余数个数小于n/m的数，把数组变化，同时更新对应值。注意如果找不到大于该数的且余数个数小于n/m的数，就是小于它的第一个数。具体细节见代码。主要熟悉了set和auto的用法。后又参考大佬的代码，他是边读边处理的结果是一样的。代码能力还有待提高啊。

我的代码：

#include<iostream>
#include<algorithm>
#include<cstdio>
#include<cstring>
#include<string>
#include<cmath>
#include<cstdlib>
#include<ctime>
#include<set>
#include<map>
#include<stack>
#include<queue>
using namespace std;
#define ll long long
#define pb push_back
#define inf 0x3f3f3f3f
#define eps 1e-10
const int N = 3e5+10;

ll n,m,cnt[N],tp[N];
ll a[N];

int main() {
    ios::sync_with_stdio(0);cin.tie(0);
    cin>>n>>m;
    ll k=n/m;
    for(int i=1;i<=n;i++) {
        cin>>a[i];
        cnt[a[i]%m]++;
    }
    set<ll> s;
    for(ll i=0;i<m;i++)
        if(cnt[i]<k) s.insert(i);
    ll ans=0;
    for(ll i=1;i<=n;i++) {
        if(cnt[a[i]%m]<=k) continue;
        auto it=s.lower_bound(a[i]%m);
        if( it!=s.end() ) {
            ans += (*it-a[i]%m);
            cnt[a[i]%m]--;
            cnt[*it]++;
            a[i]+=(*it-a[i]%m);
            if(cnt[*it]>=k) s.erase(*it);
        }else {
            ans += (m-(a[i]%m-*s.begin()));
            cnt[a[i]%m]--;
            cnt[*s.begin()]++;
            a[i]+=(m-(a[i]%m-*s.begin()));
            if( cnt[*s.begin()]>=k) s.erase(*s.begin());
        }
    }
    cout<<ans<<endl;
    for(int i=1;i<=n;i++){
        cout<<a[i];
        if(i!=n) cout<<" ";
    }
    return 0;
}

大佬的代码：

#include<iostream>
#include<algorithm>
#include<set>
using namespace std;
typedef long long ll;
const int N=2e5+6;
set<ll>st;
//cnt数组用来存0~m-1位置的所有的余数的个数都为n/m
//arr数组用来存输入的数组，以及改变后的数组
//res用来存当前值的余数
//val用来存应该最优应该选取的余数的值。
//ans用来存需要改变的操作数 
ll n,m,cnt[N],arr[N],res,val,ans;
int main(){
    cin>>n>>m;
    for(int i=0;i<m;cnt[i++]=n/m)st.insert((ll)i);
    for(int i=0;i<n;i++){
        cin>>arr[i];
        res=arr[i]%m;
        //贪心部分 
        if(res>*st.rbegin())val=*st.begin(); 
        else val= *st.lower_bound(res);
        //贪心部分 
        ans=ans+(val-res+m)%m;
        if(!--cnt[val])st.erase(val);
        arr[i]=arr[i]+(val-res+m)%m;
    }
    cout<<ans<<endl;
    for(int i=0;i<n;i++)cout<<arr[i]<<" ";
}