B. Array K-Coloring
给定一个数组
要求对元素进行染色,满足相同元素颜色一定不同,且所有颜色要全部用完
#include<bits/stdc++.h> using namespace std; #define sc(x) scanf("%I64d",&x); #define int long long vector<int> v[5005]; int ans[5005]; signed main() { int n,k,a; sc(n); sc(k); int ma=0; for(int i=0;i<n;i++){ sc(a); v[a].push_back(i); if(v[a].size()>k){ puts("NO"); return 0; } ma=max(ma,(int)v[a].size()); } int t=0; for(int j=1;j<=k;j++) for(int i=1;i<=5000;i++){ if(v[i].size()>0){ ans[v[i][0]]=j; v[i].erase(v[i].begin()); t++; } if(n-t==k-j){ break; } } puts("YES"); for(int i=0;i<n;i++){ if(ans[i]==0){ cout<<(k--)<<' '; } else cout<<ans[i]<<' '; } }