B. Array K-Coloring

B. Array K-Coloring

给定一个数组

要求对元素进行染色，满足相同元素颜色一定不同，且所有颜色要全部用完

#include<bits/stdc++.h>
using namespace std;
#define sc(x) scanf("%I64d",&x);
#define int long long
vector<int> v[];
int ans[];
signed main()
{
    int n,k,a;
    sc(n);
    sc(k);
    int ma=;
    for(int i=;i<n;i++){
        sc(a);
        v[a].push_back(i);
        if(v[a].size()>k){
            puts("NO");
            return ;
        }
        ma=max(ma,(int)v[a].size());
    }
    int t=;

    for(int j=;j<=k;j++)
    for(int i=;i<=;i++){
        if(v[i].size()>){
            ans[v[i][]]=j;
            v[i].erase(v[i].begin());
            t++;
        }
        if(n-t==k-j){
            break;
        }
    }
    puts("YES");

    for(int i=;i<n;i++){
        if(ans[i]==){
            cout<<(k--)<<' ';
        }
        else
        cout<<ans[i]<<' ';
    }
}