B2. Wonderful Coloring - 2
time limit per test
2 seconds
memory limit per test
256 megabytes
input
standard input
output
standard output
This problem is an extension of the problem "Wonderful Coloring - 1". It has quite many differences, so you should read this statement completely.
Recently, Paul and Mary have found a new favorite sequence of integers a1,a2,…,ana1,a2,…,an. They want to paint it using pieces of chalk of kk colors. The coloring of a sequence is called wonderful if the following conditions are met:
- each element of the sequence is either painted in one of kk colors or isn't painted;
- each two elements which are painted in the same color are different (i. e. there's no two equal values painted in the same color);
- let's calculate for each of kk colors the number of elements painted in the color — all calculated numbers must be equal;
- the total number of painted elements of the sequence is the maximum among all colorings of the sequence which meet the first three conditions.
E. g. consider a sequence a=[3,1,1,1,1,10,3,10,10,2]a=[3,1,1,1,1,10,3,10,10,2] and k=3k=3. One of the wonderful colorings of the sequence is shown in the figure.
The example of a wonderful coloring of the sequence a=[3,1,1,1,1,10,3,10,10,2]a=[3,1,1,1,1,10,3,10,10,2] and k=3k=3. Note that one of the elements isn't painted.
Help Paul and Mary to find a wonderful coloring of a given sequence aa.
Input
The first line contains one integer tt (1≤t≤100001≤t≤10000) — the number of test cases. Then tt test cases follow.
Each test case consists of two lines. The first one contains two integers nn and kk (1≤n≤2⋅1051≤n≤2⋅105, 1≤k≤n1≤k≤n) — the length of a given sequence and the number of colors, respectively. The second one contains nn integers a1,a2,…,ana1,a2,…,an (1≤ai≤n1≤ai≤n).
It is guaranteed that the sum of nn over all test cases doesn't exceed 2⋅1052⋅105.
Output
Output tt lines, each of them must contain a description of a wonderful coloring for the corresponding test case.
Each wonderful coloring must be printed as a sequence of nn integers c1,c2,…,cnc1,c2,…,cn (0≤ci≤k0≤ci≤k) separated by spaces where
- ci=0ci=0, if ii-th element isn't painted;
- ci>0ci>0, if ii-th element is painted in the cici-th color.
Remember that you need to maximize the total count of painted elements for the wonderful coloring. If there are multiple solutions, print any one.
Example
input
Copy
6
10 3
3 1 1 1 1 10 3 10 10 2
4 4
1 1 1 1
1 1
1
13 1
3 1 4 1 5 9 2 6 5 3 5 8 9
13 2
3 1 4 1 5 9 2 6 5 3 5 8 9
13 3
3 1 4 1 5 9 2 6 5 3 5 8 9
output
Copy
1 1 0 2 3 2 2 1 3 3
4 2 1 3
1
0 0 1 1 0 1 1 1 0 1 1 1 0
2 1 2 2 1 1 1 1 2 1 0 2 2
1 1 3 2 1 3 3 1 2 2 3 2 0
Note
In the first test case, the answer is shown in the figure in the statement. The red color has number 11, the blue color — 22, the green — 33.
题目大意
给定一个长度为 nn 的数组 aa,取出 kk 个相离的大小相同的下标集合,集合中每个下标对应的数字需不同,最大化集合的大小,并输出方案。
题目分析
首先统计每个数字出现的次数,如果大于等于 kk 那么显然每个集合分一个,否则待定。贪心地,最后将待定的数字依次 kk 个一组分配给每个集合即可,这里可以用 vector
实现。
#include <bits/stdc++.h> using namespace std; int T, n, k, x, ans[200001]; vector<int>num[200001], vec; int main() { cin >> T; while (T--) { cin >> n; cin >> k; for ( int i = 1; i <= n; ++i) { num[i].clear(); ans[i] = 0; } vec.clear(); for ( int i = 1; i <= n; ++i) { cin >> x; num[x].push_back(i); } for ( int i = 1; i <= n; ++i) { if (num[i].size() > k) for ( int j = 0; j < k; ++j) ans[num[i][j]] = j + 1; else { for ( int j = 0, up = num[i].size(); j < up; ++j) vec.push_back(num[i][j]); } } for ( int i = 0, up = vec.size(); i + k - 1 < up; i += k) { for ( int j = i; j < i + k; ++j) ans[vec[j]] = j - i + 1; } for ( int i = 1; i <= n; ++i) printf("%d ", ans[i]); puts(""); } return 0; }