-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathC.cpp
123 lines (118 loc) · 2.4 KB
/
C.cpp
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
#include <bits/stdc++.h>
using namespace std;
const int maxn = 1e5 + 10;
int a[maxn], b[maxn];
int n;
vector <int> v[maxn];
int ord[maxn];
void solve(vector <int> id) {
for(int i = 1; i <= n + 1; i++) v[i].clear();
for(int i : id) {
v[b[i]].push_back(i);
}
for(int i = 1; i <= n + 1; i++) ord[i] = i;
sort(ord + 1, ord + n + 1 + 1, [&] (int i, int j) {
return v[i].size() < v[j].size(); } );
vector <int> extra;
for(int i = 1; i < n + 1; i++) {
int x = ord[i];
for(int j : v[x]) {
a[j] = ord[i + 1];
}
if(i < n) {
int dif = v[ord[i + 1]].size() - v[ord[i]].size();
for(int j = 1; j <= dif; j++) {
extra.push_back(ord[i + 1]);
}
}
}
int last = ord[n + 1];
for(int i = 1; i <= n + 1; i++) {
int x = ord[i];
for(int j : v[x]) {
if(extra.size() >= v[last].size()) break;
extra.push_back(a[j]);
a[j] = last;
}
}
for(int j = 0; j < v[last].size(); j++) {
int x = v[last][j];
a[x] = extra[j];
}
}
void test() {
int x, y;
scanf("%d %d %d", &n, &x, &y);
for(int i = 1; i <= n + 1; i++) {
v[i].clear();
a[i] = 0;
ord[i] = i;
}
for(int i = 1; i <= n; i++) {
scanf("%d", &b[i]);
v[b[i]].push_back(i);
}
vector <int> aux;
int sz = -1;
int pt = 0, sum = 0;
for(int i = 1; i <= n + 1; i++) aux.push_back(v[i].size());
sort(aux.begin(), aux.end());
for(int j = 1; j <= n - x; j++) {
while(pt < aux.size() && aux[pt] <= j / 2) {
sum += aux[pt];
++pt;
}
int total = sum + (aux.size() - pt) * (j / 2);
if(total >= j && j >= y - x) {
sz = j;
break;
}
}
if(y > x && sz == -1) {
puts("NO");
return ;
}
int var = 0;
vector <int> id;
for(int i = 1; i <= n + 1; i++) {
if(v[i].size() == 0) var = i;
int ans = 0;
for(int j = 0; j < v[i].size() && j < sz / 2; j++) {
if(id.size() < sz) {
id.push_back(v[i][j]);
}
}
}
solve(id);
y -= x;
for(int i = 1; i <= n; i++) {
if(a[i] == 0) continue;
if(y > 0) {
y -= 1;
} else {
a[i] = var;
}
}
for(int i = 1; i <= n; i++) {
if(a[i] != 0) continue;
if(x > 0) {
a[i] = b[i];
x -= 1;
} else {
a[i] = var;
}
}
printf("YES\n");
for(int i = 1; i <= n; i++) {
printf("%d ", a[i]);
}
printf("\n");
}
int main() {
int t;
scanf("%d", &t);
while(t--) {
test();
}
return 0;
}