Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- #define _CRT_SECURE_NO_WARNINGS
- #include <iostream>
- #include <cstdio>
- #include <string>
- #include <algorithm>
- #include <cmath>
- #include <vector>
- #include <queue>
- #include <stack>
- #include <deque>
- #include <set>
- #include <map>
- #include <climits>
- #include <cstdlib>
- #include<time.h>
- #include<iomanip>
- using namespace std;
- const int N = 2e5 + 7;
- long long t, I, i, n, ed, cnt, k;
- long long a[N];
- long long num[N], col[N], ans[N];
- int main() {
- std::ios::sync_with_stdio(false);
- cin.tie(0);
- cout.tie(0);
- cin >> t;
- for (I = 0; I < t; I++) {
- cin >> n >> k;
- ed = 0;
- for (i = 1; i <= n; i++) {
- num[i] = 0;
- col[i] = 0;
- ans[i] = 0;
- }
- ans[0] = 0;
- for (i = 0; i < n; i++) {
- cin >> a[i];
- num[a[i]]++;
- }
- for (i = 1; i <= n; i++) {
- if (num[i] < k) {
- ed += num[i];
- }
- if (num[i] >= k) {
- num[i] = -1;
- col[i] = 1;
- }
- }
- ed -= ed % k;
- for (i = 0; i < n; i++) {
- if (num[a[i]] == -1) {
- if (col[a[i]] <= k) {
- ans[i] = col[a[i]];
- col[a[i]]++;
- }
- }
- else {
- if (ed > 0) {
- ans[i] = ed % k + 1;
- ed--;
- }
- }
- }
- for (i = 0; i < n; i++) {
- cout << ans[i] << " ";
- }
- cout << "\n";
- }
- }
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement