CF B2. Wonderful Coloring - 2

B2. Wonderful Coloring - 2

题目

传送门

大意为给一个数列，和k个颜色 ，对这个数列进行染色 ，使得满足以下规定：

1. 每种颜色染色数量同样多
2. 相同的数字只能染不同的颜色，相同的颜色只能染不同的数字，换句话说，不能相同数字染相同颜色。比如不能出现多个红色的1。
3. 可以有不染色的数字存在，但要保证染色的数量最大

扯淡

一开始写的模拟很复杂，然后一直WA，看了大佬的题解发现，还是分类的边界和关键属性没抓对，逻辑性还是不够，无法抽出一个简洁的思路，导致后面越改越乱。

思路

显然如果同一个数字出现次数大于或等于k 可以把这个数字用所有颜色染一遍，并且之后再有这个数字出现也没用了（由题面规定2）。并且，为了满足规定1，最后染出来的色块总数一定是一个k的整数倍，余下的都是不染色的。

因此，应该根据数字出现次数 是否大于等于k为分类标准。并且找到染色最大数量，逐一进行染色即可。

在代码实现方面，因为要知道某数字的出现所有下标，且数字不连续，开一个map 建立一个 数字 -> idx 的映射会很方便。

代码实现

#include <bits/stdc++.h>
using namespace std;
const int N = 2e5 + 4;
int arr[N];

int main()
{
    ios::sync_with_stdio(0); //加了 puts也不能用
    int T;
    cin >> T;
    while (T--)
    {
        int n, k, ans[N] = {}, sum = 0;
        map<int, vector<int>> mp; // 数字-> idx 
        cin >> n >> k;
        for (int i = 0; i < n; i++)
            cin >> arr[i];
        for (int i = 0; i < n; i++)
            if (mp[arr[i]].size() < k)//多于k的没用了，不要
                mp[arr[i]].push_back(i);
        for (auto i : mp)//看看最大能染的色块数量
            sum += i.second.size(); //mp[i].size()
        sum -= sum % k;//多余的不涂色 保证sum是k的整数倍
        int col = 0;
        for (auto a : mp)
            for (auto b : a.second)
            {
                ans[b] = ++col;
                col %= k;
                if (--sum == 0)
                    goto flag;//多样例问题还是很好用的 稍微优化下常数
            }
    flag:
        for (int i = 0; i < n; i++)
            cout << ans[i] << ' ';
        cout << endl;
    }
    return 0;
}