B2. Wonderful Coloring - 2

链接：Problem - 1551B2 - Codeforces

题意：有m个颜色，要求每种颜色内的数字各不相同，问，颜色的最大长度多少。

题解:  判断每个数字的个数，如果大于m，那么最大长度就加一，不大于就都加到一起，看看最后能凑几对。

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef pair<ll,ll> pll;
const int N=2e5+1;
ll a[N],p[N];
ll vis[N];
vector<ll> r,sum[N];
ll ans[N];
bool cmp(ll a,ll b){
  return sum[a]<sum[b];
}
signed main(){
  ios::sync_with_stdio(false);
  cin.tie(0);
  ll t;cin>>t;
  while(t--){
    ll n,m;cin>>n>>m;
    for(ll i=0;i<=n;i++){
      sum[i].clear();ans[i]=0;
    }
    r.clear();
    for(ll i=1;i<=n;i++){
      cin>>a[i];sum[a[i]].push_back(i);
    }
    for(ll i=1;i<=n;i++){
      if(!sum[i].size()) continue;
      if(sum[i].size()>=m){
           for(ll j=0;j<m;j++)
           ans[sum[i][j]]=j+1;
      }
      else 
          for(ll j=0;j<sum[i].size();j++){
             r.push_back(sum[i][j]);
          }
    }
    if(r.size()>=m){ 
    for(ll i=0;i<r.size()&&r.size()-i>=m;i+=m){
      for(ll j=0;j<m;j++) ans[r[i+j]]=j+1;
    }
    }
    for(ll i=1;i<=n;i++){
      cout<<ans[i]<<" ";
    }
    cout<<endl;
  }
}