CodeForces - 1102B Array K-Coloring

Godgrey

于 2019-04-08 22:42:49 发布

阅读量234

点赞数

本文链接：https://blog.youkuaiyun.com/qq_41476525/article/details/89113329

版权

博客详细介绍了CodeForces中1102B问题的解决方案。问题要求使用K种颜色对给定的n个数字进行染色，每个数字必须用到且只能用一种颜色，相同数字不能使用同色。若相同数字数量大于K，则情况不可能。博主给出了当数字数量不大于K时的染色策略，首先遍历1-K颜色，然后确保后续颜色未被已染色数字使用。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题意：给n个数字，和k种颜色，用k种颜色填这n个数字，每种颜色至少要用到一次同时相同的数字不能用同一种颜色。

题解：如果相同的数字大于k则不可能，小于k先从1-k遍历输出一遍，同时标记一下各个数字已经用过什么颜色了，再在后面开始输出每个数字没输出过的颜色（不先输出一遍1-k直接后面一步也可以。

#include <iostream>
#include <bits/stdc++.h>
using namespace std;
bool color[5005][5005];
int main()
{
    ios::sync_with_stdio(false);
    int n,k;
    int a[5010],b[5010];
    cin>>n>>k;
    memset(b,0,sizeof(b));
    memset(color,0,sizeof(color));
    int imax=0;
    for(int i=0;i<n;i++)
    {
        cin>>a[i];
        b[a[i]]++;
        imax=max(imax,b[a[i]]);
    }
    if(imax>k||n<k)
        cout<<"NO\n";
    else
    {
        cout<<"YES\n";
        cout<<"1";
        color[a[0]][1]=true;
        for(int i=1;i<k;i++)
        {
            cout<<' '<<i+1;
            color[a[i]][i+1]=true;
        }
        for(int i=k;i<n;i++)
        {
            for(int j=1;j<=k;j++)
            {
                if(color[a[i]][j]==false)
                {
                    cout<<' '<<j;
                    color[a[i]][j]=true;
                    break;
                }
            }
        }
        cout<<endl;
    }
	return 0;
}