标签：Coloring ma puts int ans sc Array

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]<<' ';
    }
}

标签：Coloring,ma,puts,int,ans,sc,Array
来源： https://www.cnblogs.com/liulex/p/11423372.html

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。