本文共 929 字，大约阅读时间需要 3 分钟。

题意:要求改变一个数组,使得模m后,结果为0,1,2,3,...,m-1都是n/m个,每次操作可以选择一个数+1,问至少执行多少次,并输出最终的数组

思路:模拟当前元素应该往哪个元素去改变

注意:std::set::lower_bound的复杂度为logN,而std::lower_bound的复杂度在set里是logn+n,原因大致是set是一颗平衡树,用普通的lower_bound会有额外的增量时间,具体的,不研究.

具体的,可以上stackoverflow 或者 cppreference 参考两个函数的API接差别.

#include
   
    #define PI acos(-1.0)#define pb push_back#define F first#define S secondusing namespace std;typedef long long ll;const int N=4e5+5;const int MOD=1e9+7;int a[N],sum[N];int cnt[N];set
    
      st;int main(void){    ios::sync_with_stdio(false);    cin.tie(0);cout.tie(0);    int n,m;    cin >>n >> m;    for(int i=1;i<=m;i++)   st.insert(i-1);    const int need=n/m;//    cout << need << endl;    ll ans=0;    for(int i=1;i<=n;i++){        cin >> a[i];        int mo=a[i]%m;        int dest;        if(mo>*st.rbegin())    dest=*st.begin();        else    dest=*st.lower_bound(mo);//        cout <
     
      <<"~"<
      
       <<" "<
       
        <
       
      
     
    
   

转载地址：http://ufkiz.baihongyu.com/