Array K-Coloring - codeforce

Array K-Coloring - codeforce_第1张图片

题意

\(n\)个位置染色,有\(k\)种颜色。每个位置有一个数字\(a_i\),注意:数字相同的位置的颜色不能相同,然后\(k\)种颜色都要用上。随便输出一种染色方案。

题解

排序,然后依次染色就行。

代码

const int N = 100005;

int n, k;
int cnt[N], ans[N];
P p[6000];

int main()
{
    cin >> n >> k;
    Rep(i, 1, n) {
        cin >> p[i].first;
        p[i].second = i;
        cnt[p[i].first]++;
    }

    Rep(i, 1, n) if (cnt[p[i].first] > k) {
        puts("NO");
        return 0;
    }

    sort(p + 1, p + n + 1);

    int t = 0;
    Rep(i, 1, n) {
        ans[p[i].second] = ++t;
        t %= k;
    }

    puts("YES");
    Rep(i, 1, n) cout << ans[i] << " ";
    cout << endl;

    return 0;
}

转载于:https://www.cnblogs.com/zgglj-com/p/10252555.html

你可能感兴趣的:(Array K-Coloring - codeforce)