CodeForces - 1102B Array K-Coloring

题意：给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