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
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
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 s;
    for(ll i=0;i=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<


大佬的代码:

#include
#include
#include
using namespace std;
typedef long long ll;
const int N=2e5+6;
setst;
//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>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<

你可能感兴趣的:(ACM,codefoces,算法,二分,思维)