#include<bits/stdc++.h>
#include "split.h"
#define F first
#define S second
#define PB push_back
#define sz(s) (int(s.size()))
#define bit(n, k) (((n)>>(k))&1)
using namespace std;
typedef long long ll;
typedef pair<int, int> pii;
const int maxn = 1e5 + 10;
struct DSU{
int pr[maxn];
DSU(){
memset(pr, -1, sizeof pr);
}
int fnd(int u){
return pr[u] < 0 ? u : pr[u] = fnd(pr[u]);
}
void mrg(int a, int b){
if( (a = fnd(a)) == (b = fnd(b)) )
return;
if(pr[a] > pr[b])
swap(a, b);
pr[a]+= pr[b];
pr[b] = a;
}
int size(int u){
return -pr[fnd(u)];
}
};
DSU dsu;
vector<int> v[maxn];
int pr[maxn], SZ[maxn];
bool mark[maxn];
int col[maxn];
void dfs(int u){
SZ[u] = 1;
mark[u] = 1;
for(int y : v[u]){
if(!mark[y])
pr[y] = u, dfs(y), SZ[u]+= SZ[y];
}
}
int color(int u, int cnt, int c){
assert(cnt);
cnt--;
col[u] = c;
for(int y : v[u]){
if(mark[y] == mark[u] && col[y] != c && cnt)
cnt = color(y, cnt, c);
}
return cnt;
}
vector<int> find_split(int n, int a, int b, int c, vector<int> p, vector<int> q){
vector<pii> SZS;
SZS.PB({a, 1});
SZS.PB({b, 2});
SZS.PB({c, 3});
sort(SZS.begin(), SZS.end());
int A = SZS[0].F;
auto mkay = [&](){
int cnt = 0, ZR = -1, ON = -1;
for(int i = 0; i < n; i++){
if(mark[i] == 0)
ZR = i;
else
ON = i;
cnt+= mark[i];
}
assert(A <= cnt && cnt <= n-A);
if(cnt > SZS[1].F)
swap(ZR, ON);
color(ON, SZS[0].F, SZS[0].S);
color(ZR, SZS[1].F, SZS[1].S);
vector<int> ans;
for(int i = 0; i < n; i++){
if(col[i] == 0)
col[i] = SZS[2].S;
ans.PB(col[i]);
}
return ans;
};
int m = sz(p);
for(int i = 0; i < m; i++){
v[p[i]].PB(q[i]);
v[q[i]].PB(p[i]);
}
dfs(0);
for(int u = 0; u < n; u++){
if(A <= SZ[u] && SZ[u] <= n-A){
for(int i = 0; i < n; i++)
v[i].clear();
for(int i = 1; i < n; i++)
v[i].PB(pr[i]), v[pr[i]].PB(i);
memset(mark, 0, sizeof mark);
mark[pr[u]] = 1;
dfs(u);
mark[pr[u]] = 0;
return mkay();
}
}
int root = 0;
for(int i = 0; i < n; i++){
if(SZ[i] > n-A && SZ[root] > SZ[i])
root = i;
}
for(int i = 1; i < n; i++){
if(i != root && pr[i] != root){
dsu.mrg(i, pr[i]);
}
}
for(int i = 0; i < m; i++){
if(p[i] == root || q[i] == root)
continue;
dsu.mrg(p[i], q[i]);
if(dsu.size(p[i]) >= A){
memset(mark, 0, sizeof mark);
for(int j = 0; j < n; j++){
if(dsu.fnd(j) == dsu.fnd(p[i]))
mark[j] = 1;
}
return mkay();
}
}
vector<int> ans(n);
return ans;
}
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
2 ms |
3200 KB |
ok, correct split |
2 |
Correct |
3 ms |
3200 KB |
ok, correct split |
3 |
Correct |
2 ms |
3200 KB |
ok, correct split |
4 |
Correct |
3 ms |
3200 KB |
ok, correct split |
5 |
Correct |
3 ms |
3200 KB |
ok, correct split |
6 |
Correct |
4 ms |
3200 KB |
ok, correct split |
7 |
Correct |
167 ms |
15336 KB |
ok, correct split |
8 |
Correct |
96 ms |
13848 KB |
ok, correct split |
9 |
Correct |
106 ms |
13496 KB |
ok, correct split |
10 |
Correct |
104 ms |
15512 KB |
ok, correct split |
11 |
Correct |
121 ms |
15520 KB |
ok, correct split |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
3 ms |
3328 KB |
ok, correct split |
2 |
Correct |
3 ms |
3200 KB |
ok, correct split |
3 |
Correct |
2 ms |
3200 KB |
ok, correct split |
4 |
Correct |
148 ms |
14556 KB |
ok, correct split |
5 |
Correct |
99 ms |
10872 KB |
ok, correct split |
6 |
Correct |
110 ms |
15352 KB |
ok, correct split |
7 |
Correct |
131 ms |
13860 KB |
ok, correct split |
8 |
Correct |
172 ms |
14200 KB |
ok, correct split |
9 |
Correct |
118 ms |
10872 KB |
ok, correct split |
10 |
Correct |
64 ms |
10864 KB |
ok, correct split |
11 |
Correct |
65 ms |
10996 KB |
ok, correct split |
12 |
Correct |
73 ms |
11224 KB |
ok, correct split |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
3 ms |
3200 KB |
ok, correct split |
2 |
Correct |
103 ms |
10872 KB |
ok, correct split |
3 |
Correct |
32 ms |
6392 KB |
ok, correct split |
4 |
Correct |
2 ms |
3200 KB |
ok, correct split |
5 |
Correct |
112 ms |
12408 KB |
ok, correct split |
6 |
Correct |
136 ms |
12348 KB |
ok, correct split |
7 |
Correct |
139 ms |
12172 KB |
ok, correct split |
8 |
Correct |
115 ms |
12884 KB |
ok, correct split |
9 |
Correct |
121 ms |
12024 KB |
ok, correct split |
10 |
Correct |
26 ms |
5500 KB |
ok, no valid answer |
11 |
Correct |
43 ms |
6656 KB |
ok, no valid answer |
12 |
Correct |
77 ms |
10484 KB |
ok, no valid answer |
13 |
Correct |
87 ms |
10360 KB |
ok, no valid answer |
14 |
Correct |
61 ms |
10608 KB |
ok, no valid answer |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
2 ms |
3200 KB |
ok, correct split |
2 |
Correct |
2 ms |
3072 KB |
ok, no valid answer |
3 |
Correct |
3 ms |
3232 KB |
ok, correct split |
4 |
Correct |
3 ms |
3200 KB |
ok, correct split |
5 |
Correct |
2 ms |
3124 KB |
ok, correct split |
6 |
Correct |
2 ms |
3200 KB |
ok, correct split |
7 |
Correct |
3 ms |
3200 KB |
ok, correct split |
8 |
Correct |
2 ms |
3200 KB |
ok, correct split |
9 |
Correct |
5 ms |
3456 KB |
ok, correct split |
10 |
Correct |
5 ms |
3488 KB |
ok, correct split |
11 |
Correct |
3 ms |
3200 KB |
ok, correct split |
12 |
Correct |
5 ms |
3456 KB |
ok, correct split |
13 |
Correct |
2 ms |
3124 KB |
ok, correct split |
14 |
Correct |
3 ms |
3200 KB |
ok, correct split |
15 |
Correct |
2 ms |
3200 KB |
ok, correct split |
16 |
Correct |
2 ms |
3200 KB |
ok, correct split |
17 |
Correct |
3 ms |
3200 KB |
ok, correct split |
18 |
Correct |
2 ms |
3200 KB |
ok, correct split |
19 |
Correct |
3 ms |
3200 KB |
ok, correct split |
20 |
Correct |
3 ms |
3328 KB |
ok, correct split |
21 |
Correct |
4 ms |
3456 KB |
ok, correct split |
22 |
Correct |
4 ms |
3456 KB |
ok, correct split |
23 |
Correct |
4 ms |
3456 KB |
ok, correct split |
24 |
Correct |
8 ms |
3392 KB |
ok, correct split |
25 |
Correct |
4 ms |
3432 KB |
ok, correct split |
26 |
Correct |
5 ms |
3456 KB |
ok, correct split |
27 |
Correct |
5 ms |
3456 KB |
ok, correct split |
28 |
Correct |
5 ms |
3456 KB |
ok, correct split |
29 |
Correct |
3 ms |
3200 KB |
ok, correct split |
30 |
Correct |
5 ms |
3456 KB |
ok, correct split |
31 |
Correct |
3 ms |
3200 KB |
ok, correct split |
32 |
Correct |
4 ms |
3200 KB |
ok, correct split |
33 |
Correct |
3 ms |
3232 KB |
ok, correct split |
34 |
Correct |
5 ms |
3360 KB |
ok, correct split |
35 |
Correct |
5 ms |
3328 KB |
ok, correct split |
36 |
Correct |
4 ms |
3328 KB |
ok, correct split |
37 |
Correct |
5 ms |
3456 KB |
ok, correct split |
38 |
Correct |
5 ms |
3456 KB |
ok, correct split |
39 |
Correct |
5 ms |
3456 KB |
ok, correct split |
40 |
Correct |
6 ms |
3488 KB |
ok, correct split |
41 |
Correct |
4 ms |
3360 KB |
ok, correct split |
42 |
Correct |
4 ms |
3328 KB |
ok, correct split |
43 |
Correct |
5 ms |
3456 KB |
ok, correct split |
44 |
Correct |
5 ms |
3456 KB |
ok, correct split |
45 |
Correct |
5 ms |
3456 KB |
ok, correct split |
46 |
Correct |
4 ms |
3456 KB |
ok, correct split |
47 |
Correct |
4 ms |
3328 KB |
ok, no valid answer |
48 |
Correct |
4 ms |
3328 KB |
ok, correct split |
49 |
Correct |
4 ms |
3456 KB |
ok, correct split |
50 |
Correct |
2 ms |
3072 KB |
ok, no valid answer |
51 |
Correct |
3 ms |
3072 KB |
ok, no valid answer |
52 |
Correct |
4 ms |
3200 KB |
ok, no valid answer |
53 |
Correct |
6 ms |
3328 KB |
ok, no valid answer |
54 |
Correct |
8 ms |
3200 KB |
ok, no valid answer |
55 |
Correct |
7 ms |
3328 KB |
ok, no valid answer |
56 |
Correct |
6 ms |
3328 KB |
ok, no valid answer |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
2 ms |
3200 KB |
ok, correct split |
2 |
Correct |
3 ms |
3200 KB |
ok, correct split |
3 |
Correct |
2 ms |
3200 KB |
ok, correct split |
4 |
Correct |
3 ms |
3200 KB |
ok, correct split |
5 |
Correct |
3 ms |
3200 KB |
ok, correct split |
6 |
Correct |
4 ms |
3200 KB |
ok, correct split |
7 |
Correct |
167 ms |
15336 KB |
ok, correct split |
8 |
Correct |
96 ms |
13848 KB |
ok, correct split |
9 |
Correct |
106 ms |
13496 KB |
ok, correct split |
10 |
Correct |
104 ms |
15512 KB |
ok, correct split |
11 |
Correct |
121 ms |
15520 KB |
ok, correct split |
12 |
Correct |
3 ms |
3328 KB |
ok, correct split |
13 |
Correct |
3 ms |
3200 KB |
ok, correct split |
14 |
Correct |
2 ms |
3200 KB |
ok, correct split |
15 |
Correct |
148 ms |
14556 KB |
ok, correct split |
16 |
Correct |
99 ms |
10872 KB |
ok, correct split |
17 |
Correct |
110 ms |
15352 KB |
ok, correct split |
18 |
Correct |
131 ms |
13860 KB |
ok, correct split |
19 |
Correct |
172 ms |
14200 KB |
ok, correct split |
20 |
Correct |
118 ms |
10872 KB |
ok, correct split |
21 |
Correct |
64 ms |
10864 KB |
ok, correct split |
22 |
Correct |
65 ms |
10996 KB |
ok, correct split |
23 |
Correct |
73 ms |
11224 KB |
ok, correct split |
24 |
Correct |
3 ms |
3200 KB |
ok, correct split |
25 |
Correct |
103 ms |
10872 KB |
ok, correct split |
26 |
Correct |
32 ms |
6392 KB |
ok, correct split |
27 |
Correct |
2 ms |
3200 KB |
ok, correct split |
28 |
Correct |
112 ms |
12408 KB |
ok, correct split |
29 |
Correct |
136 ms |
12348 KB |
ok, correct split |
30 |
Correct |
139 ms |
12172 KB |
ok, correct split |
31 |
Correct |
115 ms |
12884 KB |
ok, correct split |
32 |
Correct |
121 ms |
12024 KB |
ok, correct split |
33 |
Correct |
26 ms |
5500 KB |
ok, no valid answer |
34 |
Correct |
43 ms |
6656 KB |
ok, no valid answer |
35 |
Correct |
77 ms |
10484 KB |
ok, no valid answer |
36 |
Correct |
87 ms |
10360 KB |
ok, no valid answer |
37 |
Correct |
61 ms |
10608 KB |
ok, no valid answer |
38 |
Correct |
2 ms |
3200 KB |
ok, correct split |
39 |
Correct |
2 ms |
3072 KB |
ok, no valid answer |
40 |
Correct |
3 ms |
3232 KB |
ok, correct split |
41 |
Correct |
3 ms |
3200 KB |
ok, correct split |
42 |
Correct |
2 ms |
3124 KB |
ok, correct split |
43 |
Correct |
2 ms |
3200 KB |
ok, correct split |
44 |
Correct |
3 ms |
3200 KB |
ok, correct split |
45 |
Correct |
2 ms |
3200 KB |
ok, correct split |
46 |
Correct |
5 ms |
3456 KB |
ok, correct split |
47 |
Correct |
5 ms |
3488 KB |
ok, correct split |
48 |
Correct |
3 ms |
3200 KB |
ok, correct split |
49 |
Correct |
5 ms |
3456 KB |
ok, correct split |
50 |
Correct |
2 ms |
3124 KB |
ok, correct split |
51 |
Correct |
3 ms |
3200 KB |
ok, correct split |
52 |
Correct |
2 ms |
3200 KB |
ok, correct split |
53 |
Correct |
2 ms |
3200 KB |
ok, correct split |
54 |
Correct |
3 ms |
3200 KB |
ok, correct split |
55 |
Correct |
2 ms |
3200 KB |
ok, correct split |
56 |
Correct |
3 ms |
3200 KB |
ok, correct split |
57 |
Correct |
3 ms |
3328 KB |
ok, correct split |
58 |
Correct |
4 ms |
3456 KB |
ok, correct split |
59 |
Correct |
4 ms |
3456 KB |
ok, correct split |
60 |
Correct |
4 ms |
3456 KB |
ok, correct split |
61 |
Correct |
8 ms |
3392 KB |
ok, correct split |
62 |
Correct |
4 ms |
3432 KB |
ok, correct split |
63 |
Correct |
5 ms |
3456 KB |
ok, correct split |
64 |
Correct |
5 ms |
3456 KB |
ok, correct split |
65 |
Correct |
5 ms |
3456 KB |
ok, correct split |
66 |
Correct |
3 ms |
3200 KB |
ok, correct split |
67 |
Correct |
5 ms |
3456 KB |
ok, correct split |
68 |
Correct |
3 ms |
3200 KB |
ok, correct split |
69 |
Correct |
4 ms |
3200 KB |
ok, correct split |
70 |
Correct |
3 ms |
3232 KB |
ok, correct split |
71 |
Correct |
5 ms |
3360 KB |
ok, correct split |
72 |
Correct |
5 ms |
3328 KB |
ok, correct split |
73 |
Correct |
4 ms |
3328 KB |
ok, correct split |
74 |
Correct |
5 ms |
3456 KB |
ok, correct split |
75 |
Correct |
5 ms |
3456 KB |
ok, correct split |
76 |
Correct |
5 ms |
3456 KB |
ok, correct split |
77 |
Correct |
6 ms |
3488 KB |
ok, correct split |
78 |
Correct |
4 ms |
3360 KB |
ok, correct split |
79 |
Correct |
4 ms |
3328 KB |
ok, correct split |
80 |
Correct |
5 ms |
3456 KB |
ok, correct split |
81 |
Correct |
5 ms |
3456 KB |
ok, correct split |
82 |
Correct |
5 ms |
3456 KB |
ok, correct split |
83 |
Correct |
4 ms |
3456 KB |
ok, correct split |
84 |
Correct |
4 ms |
3328 KB |
ok, no valid answer |
85 |
Correct |
4 ms |
3328 KB |
ok, correct split |
86 |
Correct |
4 ms |
3456 KB |
ok, correct split |
87 |
Correct |
2 ms |
3072 KB |
ok, no valid answer |
88 |
Correct |
3 ms |
3072 KB |
ok, no valid answer |
89 |
Correct |
4 ms |
3200 KB |
ok, no valid answer |
90 |
Correct |
6 ms |
3328 KB |
ok, no valid answer |
91 |
Correct |
8 ms |
3200 KB |
ok, no valid answer |
92 |
Correct |
7 ms |
3328 KB |
ok, no valid answer |
93 |
Correct |
6 ms |
3328 KB |
ok, no valid answer |
94 |
Correct |
130 ms |
12872 KB |
ok, correct split |
95 |
Correct |
188 ms |
17400 KB |
ok, correct split |
96 |
Correct |
176 ms |
15732 KB |
ok, correct split |
97 |
Correct |
32 ms |
6392 KB |
ok, correct split |
98 |
Correct |
34 ms |
6520 KB |
ok, correct split |
99 |
Correct |
56 ms |
8832 KB |
ok, correct split |
100 |
Correct |
153 ms |
14824 KB |
ok, correct split |
101 |
Correct |
166 ms |
13556 KB |
ok, correct split |
102 |
Correct |
133 ms |
13168 KB |
ok, correct split |
103 |
Correct |
123 ms |
12948 KB |
ok, correct split |
104 |
Correct |
122 ms |
14564 KB |
ok, correct split |
105 |
Correct |
68 ms |
8056 KB |
ok, correct split |
106 |
Correct |
145 ms |
14440 KB |
ok, correct split |
107 |
Correct |
96 ms |
11000 KB |
ok, correct split |
108 |
Correct |
94 ms |
10872 KB |
ok, correct split |
109 |
Correct |
164 ms |
14260 KB |
ok, correct split |
110 |
Correct |
165 ms |
14040 KB |
ok, correct split |
111 |
Correct |
157 ms |
14192 KB |
ok, correct split |
112 |
Correct |
159 ms |
14392 KB |
ok, correct split |
113 |
Correct |
147 ms |
14388 KB |
ok, correct split |
114 |
Correct |
14 ms |
4608 KB |
ok, correct split |
115 |
Correct |
14 ms |
4352 KB |
ok, correct split |
116 |
Correct |
159 ms |
14064 KB |
ok, correct split |
117 |
Correct |
144 ms |
13936 KB |
ok, correct split |
118 |
Correct |
112 ms |
10872 KB |
ok, correct split |
119 |
Correct |
107 ms |
11000 KB |
ok, correct split |
120 |
Correct |
120 ms |
13304 KB |
ok, correct split |
121 |
Correct |
88 ms |
10332 KB |
ok, no valid answer |
122 |
Correct |
93 ms |
10280 KB |
ok, no valid answer |
123 |
Correct |
147 ms |
14200 KB |
ok, no valid answer |
124 |
Correct |
160 ms |
14252 KB |
ok, no valid answer |
125 |
Correct |
91 ms |
11576 KB |
ok, no valid answer |
126 |
Correct |
55 ms |
9844 KB |
ok, no valid answer |
127 |
Correct |
119 ms |
11988 KB |
ok, no valid answer |