Advertisement
bibaboba12345

ХУЙ

Jul 23rd, 2021
128
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 1.21 KB | None | 0 0
  1. #define _CRT_SECURE_NO_WARNINGS
  2. #include <iostream>
  3. #include <cstdio>
  4. #include <string>
  5. #include <algorithm>
  6. #include <cmath>
  7. #include <vector>
  8. #include <queue>
  9. #include <stack>
  10. #include <deque>
  11. #include <set>
  12. #include <map>
  13. #include <climits>
  14. #include <cstdlib>
  15. #include<time.h>
  16. #include<iomanip>
  17. using namespace std;
  18. const int N = 2e5 + 7;
  19. long long t, I, i, n, ed, cnt, k;
  20. long long a[N];
  21. long long num[N], col[N], ans[N];
  22. int main() {
  23. std::ios::sync_with_stdio(false);
  24. cin.tie(0);
  25. cout.tie(0);
  26. cin >> t;
  27. for (I = 0; I < t; I++) {
  28. cin >> n >> k;
  29. ed = 0;
  30. for (i = 1; i <= n; i++) {
  31. num[i] = 0;
  32. col[i] = 0;
  33. ans[i] = 0;
  34. }
  35. ans[0] = 0;
  36. for (i = 0; i < n; i++) {
  37. cin >> a[i];
  38. num[a[i]]++;
  39. }
  40. for (i = 1; i <= n; i++) {
  41. if (num[i] < k) {
  42. ed += num[i];
  43. }
  44. if (num[i] >= k) {
  45. num[i] = -1;
  46. col[i] = 1;
  47. }
  48. }
  49. ed -= ed % k;
  50.  
  51. for (i = 0; i < n; i++) {
  52. if (num[a[i]] == -1) {
  53. if (col[a[i]] <= k) {
  54. ans[i] = col[a[i]];
  55. col[a[i]]++;
  56. }
  57. }
  58. else {
  59. if (ed > 0) {
  60. ans[i] = ed % k + 1;
  61. ed--;
  62. }
  63. }
  64. }
  65. for (i = 0; i < n; i++) {
  66. cout << ans[i] << " ";
  67. }
  68. cout << "\n";
  69. }
  70. }
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement