#include <bits/stdc++.h>
#define f first
#define s second
#define ent '\n'
using namespace std;
typedef long long ll;
const int mx=2e5+12;
pair<int,int> fup[mx];
vector<int> g[mx];
set<int> e[mx];
int Ans[mx];
int par[mx];
int tin[mx];
int ver[mx];
int sz[mx];
int pos[]={1,2,3};
int timer;
void calc(int v){
sz[v]=1;
for(int to:g[v]){
if(sz[to]>0)continue;
calc(to);
par[to]=v;
e[v].insert(to);
e[to].insert(v);
sz[v]+=sz[to];
}
}
void dfs(int v,int p){
tin[v]=fup[v].f=++timer;
fup[v].s=v;
ver[tin[v]]=v;
for(int to:g[v]){
if(to==p)continue;
if(tin[to]>0){
fup[v]=min(fup[v],{tin[to],v});
}
else{
dfs(to,v);
fup[v]=min(fup[v],fup[to]);
}
}
}
void get(int v,int p,int &cnt,int t){
if(!cnt)return;
Ans[v]=pos[t];
cnt--;
if(!cnt)return;
for(int to:e[v]){
if(to!=p)get(to,v,cnt,t);
}
}
std::vector<int> find_split(int n, int a, int b, int c, std::vector<int> p, std::vector<int> q){
for(int i=0;i<p.size();i++){
g[p[i]].push_back(q[i]);
g[q[i]].push_back(p[i]);
}
if(a>b){
swap(a,b);
swap(pos[0],pos[1]);
}
if(a>c){
swap(a,c);
swap(pos[0],pos[2]);
}
if(b>c){
swap(b,c);
swap(pos[1],pos[2]);
}
int A=a,B=b,C=c;
calc(0);
dfs(0,0);
int v=-1;
for(int i=0;i<n;i++){
}
for(int i=1;i<n;i++){
if(sz[i]<a)continue;
bool ok=1;
for(int to:e[i]){
if(sz[to]>=a && tin[to]>tin[i])ok=0;
}
if(ok){
v=i;
break;
}
}
vector<int> ans(n);
if(v<0)return ans;
bool bad=0;
vector<int> edg;
for(int to:e[v]){
if(tin[to]>tin[v])edg.push_back(to);
}
for(int to:edg){
if(sz[v]-sz[to]<a){
break;
}
if(fup[to].f<tin[v]){
sz[v]-=sz[to];
e[v].erase(to);
e[to].erase(v);
e[ver[fup[to].f]].insert(fup[to].s);
e[fup[to].s].insert(ver[fup[to].f]);
}
}
e[par[v]].erase(v);
e[v].erase(par[v]);
if(sz[v]>=a && n-sz[v]>=b){
get(v,0,a,0);
get(0,0,b,1);
for(int i=0;i<n;i++){
ans[i]=Ans[i];
if(!ans[i])ans[i]=pos[2];
}
}
else if(sz[v]>=b && n-sz[v]>=a){
get(v,0,b,1);
get(0,0,a,0);
for(int i=0;i<n;i++){
ans[i]=Ans[i];
if(!ans[i])ans[i]=pos[2];
}
}
return ans;
}
Compilation message
split.cpp: In function 'std::vector<int> find_split(int, int, int, int, std::vector<int>, std::vector<int>)':
split.cpp:60:15: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
60 | for(int i=0;i<p.size();i++){
| ~^~~~~~~~~
split.cpp:76:6: warning: unused variable 'A' [-Wunused-variable]
76 | int A=a,B=b,C=c;
| ^
split.cpp:76:10: warning: unused variable 'B' [-Wunused-variable]
76 | int A=a,B=b,C=c;
| ^
split.cpp:76:14: warning: unused variable 'C' [-Wunused-variable]
76 | int A=a,B=b,C=c;
| ^
split.cpp:95:7: warning: unused variable 'bad' [-Wunused-variable]
95 | bool bad=0;
| ^~~
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
5 ms |
18776 KB |
ok, correct split |
2 |
Correct |
4 ms |
18776 KB |
ok, correct split |
3 |
Correct |
5 ms |
18780 KB |
ok, correct split |
4 |
Correct |
4 ms |
18780 KB |
ok, correct split |
5 |
Correct |
5 ms |
18868 KB |
ok, correct split |
6 |
Correct |
5 ms |
18780 KB |
ok, correct split |
7 |
Correct |
125 ms |
43076 KB |
ok, correct split |
8 |
Correct |
99 ms |
39860 KB |
ok, correct split |
9 |
Correct |
108 ms |
38892 KB |
ok, correct split |
10 |
Correct |
105 ms |
43504 KB |
ok, correct split |
11 |
Correct |
109 ms |
43776 KB |
ok, correct split |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
4 ms |
18780 KB |
ok, correct split |
2 |
Correct |
4 ms |
18780 KB |
ok, correct split |
3 |
Correct |
5 ms |
18776 KB |
ok, correct split |
4 |
Correct |
101 ms |
38612 KB |
ok, correct split |
5 |
Correct |
83 ms |
32728 KB |
ok, correct split |
6 |
Correct |
96 ms |
43600 KB |
ok, correct split |
7 |
Correct |
120 ms |
39504 KB |
ok, correct split |
8 |
Correct |
111 ms |
34908 KB |
ok, correct split |
9 |
Correct |
82 ms |
32596 KB |
ok, correct split |
10 |
Correct |
109 ms |
33040 KB |
ok, correct split |
11 |
Correct |
105 ms |
32968 KB |
ok, correct split |
12 |
Correct |
114 ms |
32856 KB |
ok, correct split |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
5 ms |
18780 KB |
ok, correct split |
2 |
Correct |
98 ms |
32520 KB |
ok, correct split |
3 |
Correct |
31 ms |
24924 KB |
ok, correct split |
4 |
Correct |
4 ms |
18776 KB |
ok, correct split |
5 |
Correct |
120 ms |
36404 KB |
ok, correct split |
6 |
Correct |
100 ms |
35924 KB |
ok, correct split |
7 |
Correct |
95 ms |
35668 KB |
ok, correct split |
8 |
Correct |
97 ms |
37348 KB |
ok, correct split |
9 |
Correct |
111 ms |
35412 KB |
ok, correct split |
10 |
Correct |
25 ms |
23756 KB |
ok, no valid answer |
11 |
Correct |
42 ms |
24668 KB |
ok, no valid answer |
12 |
Correct |
93 ms |
33232 KB |
ok, no valid answer |
13 |
Correct |
103 ms |
32820 KB |
ok, no valid answer |
14 |
Correct |
113 ms |
33692 KB |
ok, no valid answer |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
5 ms |
18780 KB |
ok, correct split |
2 |
Correct |
6 ms |
18876 KB |
ok, no valid answer |
3 |
Correct |
4 ms |
18776 KB |
ok, correct split |
4 |
Correct |
4 ms |
18912 KB |
ok, correct split |
5 |
Correct |
4 ms |
18780 KB |
ok, correct split |
6 |
Correct |
5 ms |
18780 KB |
ok, correct split |
7 |
Correct |
5 ms |
18780 KB |
ok, correct split |
8 |
Correct |
5 ms |
18780 KB |
ok, correct split |
9 |
Correct |
7 ms |
19292 KB |
ok, correct split |
10 |
Correct |
7 ms |
19292 KB |
ok, correct split |
11 |
Correct |
4 ms |
18780 KB |
ok, correct split |
12 |
Correct |
7 ms |
19292 KB |
ok, correct split |
13 |
Correct |
4 ms |
18780 KB |
ok, correct split |
14 |
Correct |
4 ms |
18780 KB |
ok, correct split |
15 |
Correct |
5 ms |
18780 KB |
ok, correct split |
16 |
Correct |
5 ms |
18780 KB |
ok, correct split |
17 |
Correct |
4 ms |
18780 KB |
ok, correct split |
18 |
Correct |
8 ms |
18876 KB |
ok, correct split |
19 |
Correct |
5 ms |
18780 KB |
ok, correct split |
20 |
Correct |
5 ms |
19036 KB |
ok, correct split |
21 |
Correct |
6 ms |
19292 KB |
ok, correct split |
22 |
Correct |
5 ms |
19292 KB |
ok, correct split |
23 |
Correct |
6 ms |
19292 KB |
ok, correct split |
24 |
Correct |
5 ms |
19292 KB |
ok, correct split |
25 |
Correct |
6 ms |
19288 KB |
ok, correct split |
26 |
Correct |
7 ms |
19608 KB |
ok, correct split |
27 |
Correct |
7 ms |
19292 KB |
ok, correct split |
28 |
Correct |
6 ms |
19288 KB |
ok, correct split |
29 |
Correct |
5 ms |
18780 KB |
ok, correct split |
30 |
Correct |
7 ms |
19292 KB |
ok, correct split |
31 |
Correct |
5 ms |
18892 KB |
ok, correct split |
32 |
Correct |
5 ms |
18780 KB |
ok, correct split |
33 |
Correct |
5 ms |
18780 KB |
ok, correct split |
34 |
Correct |
6 ms |
19292 KB |
ok, correct split |
35 |
Correct |
6 ms |
19292 KB |
ok, correct split |
36 |
Correct |
8 ms |
19032 KB |
ok, correct split |
37 |
Correct |
6 ms |
19292 KB |
ok, correct split |
38 |
Correct |
7 ms |
19288 KB |
ok, correct split |
39 |
Correct |
7 ms |
19288 KB |
ok, correct split |
40 |
Correct |
7 ms |
19548 KB |
ok, correct split |
41 |
Correct |
6 ms |
19132 KB |
ok, correct split |
42 |
Correct |
6 ms |
19036 KB |
ok, correct split |
43 |
Correct |
7 ms |
19288 KB |
ok, correct split |
44 |
Correct |
8 ms |
19292 KB |
ok, correct split |
45 |
Correct |
7 ms |
19292 KB |
ok, correct split |
46 |
Correct |
7 ms |
19544 KB |
ok, correct split |
47 |
Correct |
6 ms |
19292 KB |
ok, no valid answer |
48 |
Correct |
7 ms |
19292 KB |
ok, correct split |
49 |
Correct |
6 ms |
19292 KB |
ok, correct split |
50 |
Correct |
5 ms |
18780 KB |
ok, no valid answer |
51 |
Correct |
4 ms |
18780 KB |
ok, no valid answer |
52 |
Correct |
6 ms |
19120 KB |
ok, no valid answer |
53 |
Correct |
7 ms |
19292 KB |
ok, no valid answer |
54 |
Correct |
7 ms |
19292 KB |
ok, no valid answer |
55 |
Correct |
8 ms |
19124 KB |
ok, no valid answer |
56 |
Correct |
6 ms |
19292 KB |
ok, no valid answer |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
5 ms |
18776 KB |
ok, correct split |
2 |
Correct |
4 ms |
18776 KB |
ok, correct split |
3 |
Correct |
5 ms |
18780 KB |
ok, correct split |
4 |
Correct |
4 ms |
18780 KB |
ok, correct split |
5 |
Correct |
5 ms |
18868 KB |
ok, correct split |
6 |
Correct |
5 ms |
18780 KB |
ok, correct split |
7 |
Correct |
125 ms |
43076 KB |
ok, correct split |
8 |
Correct |
99 ms |
39860 KB |
ok, correct split |
9 |
Correct |
108 ms |
38892 KB |
ok, correct split |
10 |
Correct |
105 ms |
43504 KB |
ok, correct split |
11 |
Correct |
109 ms |
43776 KB |
ok, correct split |
12 |
Correct |
4 ms |
18780 KB |
ok, correct split |
13 |
Correct |
4 ms |
18780 KB |
ok, correct split |
14 |
Correct |
5 ms |
18776 KB |
ok, correct split |
15 |
Correct |
101 ms |
38612 KB |
ok, correct split |
16 |
Correct |
83 ms |
32728 KB |
ok, correct split |
17 |
Correct |
96 ms |
43600 KB |
ok, correct split |
18 |
Correct |
120 ms |
39504 KB |
ok, correct split |
19 |
Correct |
111 ms |
34908 KB |
ok, correct split |
20 |
Correct |
82 ms |
32596 KB |
ok, correct split |
21 |
Correct |
109 ms |
33040 KB |
ok, correct split |
22 |
Correct |
105 ms |
32968 KB |
ok, correct split |
23 |
Correct |
114 ms |
32856 KB |
ok, correct split |
24 |
Correct |
5 ms |
18780 KB |
ok, correct split |
25 |
Correct |
98 ms |
32520 KB |
ok, correct split |
26 |
Correct |
31 ms |
24924 KB |
ok, correct split |
27 |
Correct |
4 ms |
18776 KB |
ok, correct split |
28 |
Correct |
120 ms |
36404 KB |
ok, correct split |
29 |
Correct |
100 ms |
35924 KB |
ok, correct split |
30 |
Correct |
95 ms |
35668 KB |
ok, correct split |
31 |
Correct |
97 ms |
37348 KB |
ok, correct split |
32 |
Correct |
111 ms |
35412 KB |
ok, correct split |
33 |
Correct |
25 ms |
23756 KB |
ok, no valid answer |
34 |
Correct |
42 ms |
24668 KB |
ok, no valid answer |
35 |
Correct |
93 ms |
33232 KB |
ok, no valid answer |
36 |
Correct |
103 ms |
32820 KB |
ok, no valid answer |
37 |
Correct |
113 ms |
33692 KB |
ok, no valid answer |
38 |
Correct |
5 ms |
18780 KB |
ok, correct split |
39 |
Correct |
6 ms |
18876 KB |
ok, no valid answer |
40 |
Correct |
4 ms |
18776 KB |
ok, correct split |
41 |
Correct |
4 ms |
18912 KB |
ok, correct split |
42 |
Correct |
4 ms |
18780 KB |
ok, correct split |
43 |
Correct |
5 ms |
18780 KB |
ok, correct split |
44 |
Correct |
5 ms |
18780 KB |
ok, correct split |
45 |
Correct |
5 ms |
18780 KB |
ok, correct split |
46 |
Correct |
7 ms |
19292 KB |
ok, correct split |
47 |
Correct |
7 ms |
19292 KB |
ok, correct split |
48 |
Correct |
4 ms |
18780 KB |
ok, correct split |
49 |
Correct |
7 ms |
19292 KB |
ok, correct split |
50 |
Correct |
4 ms |
18780 KB |
ok, correct split |
51 |
Correct |
4 ms |
18780 KB |
ok, correct split |
52 |
Correct |
5 ms |
18780 KB |
ok, correct split |
53 |
Correct |
5 ms |
18780 KB |
ok, correct split |
54 |
Correct |
4 ms |
18780 KB |
ok, correct split |
55 |
Correct |
8 ms |
18876 KB |
ok, correct split |
56 |
Correct |
5 ms |
18780 KB |
ok, correct split |
57 |
Correct |
5 ms |
19036 KB |
ok, correct split |
58 |
Correct |
6 ms |
19292 KB |
ok, correct split |
59 |
Correct |
5 ms |
19292 KB |
ok, correct split |
60 |
Correct |
6 ms |
19292 KB |
ok, correct split |
61 |
Correct |
5 ms |
19292 KB |
ok, correct split |
62 |
Correct |
6 ms |
19288 KB |
ok, correct split |
63 |
Correct |
7 ms |
19608 KB |
ok, correct split |
64 |
Correct |
7 ms |
19292 KB |
ok, correct split |
65 |
Correct |
6 ms |
19288 KB |
ok, correct split |
66 |
Correct |
5 ms |
18780 KB |
ok, correct split |
67 |
Correct |
7 ms |
19292 KB |
ok, correct split |
68 |
Correct |
5 ms |
18892 KB |
ok, correct split |
69 |
Correct |
5 ms |
18780 KB |
ok, correct split |
70 |
Correct |
5 ms |
18780 KB |
ok, correct split |
71 |
Correct |
6 ms |
19292 KB |
ok, correct split |
72 |
Correct |
6 ms |
19292 KB |
ok, correct split |
73 |
Correct |
8 ms |
19032 KB |
ok, correct split |
74 |
Correct |
6 ms |
19292 KB |
ok, correct split |
75 |
Correct |
7 ms |
19288 KB |
ok, correct split |
76 |
Correct |
7 ms |
19288 KB |
ok, correct split |
77 |
Correct |
7 ms |
19548 KB |
ok, correct split |
78 |
Correct |
6 ms |
19132 KB |
ok, correct split |
79 |
Correct |
6 ms |
19036 KB |
ok, correct split |
80 |
Correct |
7 ms |
19288 KB |
ok, correct split |
81 |
Correct |
8 ms |
19292 KB |
ok, correct split |
82 |
Correct |
7 ms |
19292 KB |
ok, correct split |
83 |
Correct |
7 ms |
19544 KB |
ok, correct split |
84 |
Correct |
6 ms |
19292 KB |
ok, no valid answer |
85 |
Correct |
7 ms |
19292 KB |
ok, correct split |
86 |
Correct |
6 ms |
19292 KB |
ok, correct split |
87 |
Correct |
5 ms |
18780 KB |
ok, no valid answer |
88 |
Correct |
4 ms |
18780 KB |
ok, no valid answer |
89 |
Correct |
6 ms |
19120 KB |
ok, no valid answer |
90 |
Correct |
7 ms |
19292 KB |
ok, no valid answer |
91 |
Correct |
7 ms |
19292 KB |
ok, no valid answer |
92 |
Correct |
8 ms |
19124 KB |
ok, no valid answer |
93 |
Correct |
6 ms |
19292 KB |
ok, no valid answer |
94 |
Correct |
104 ms |
37528 KB |
ok, correct split |
95 |
Correct |
136 ms |
43860 KB |
ok, correct split |
96 |
Correct |
124 ms |
41992 KB |
ok, correct split |
97 |
Correct |
33 ms |
25292 KB |
ok, correct split |
98 |
Correct |
32 ms |
25528 KB |
ok, correct split |
99 |
Correct |
44 ms |
27476 KB |
ok, correct split |
100 |
Correct |
133 ms |
37972 KB |
ok, correct split |
101 |
Correct |
102 ms |
36620 KB |
ok, correct split |
102 |
Correct |
106 ms |
36020 KB |
ok, correct split |
103 |
Correct |
111 ms |
35788 KB |
ok, correct split |
104 |
Correct |
138 ms |
37340 KB |
ok, correct split |
105 |
Correct |
53 ms |
27220 KB |
ok, correct split |
106 |
Correct |
204 ms |
37808 KB |
ok, correct split |
107 |
Correct |
87 ms |
33956 KB |
ok, correct split |
108 |
Correct |
87 ms |
33880 KB |
ok, correct split |
109 |
Correct |
115 ms |
37204 KB |
ok, correct split |
110 |
Correct |
106 ms |
36716 KB |
ok, correct split |
111 |
Correct |
109 ms |
37068 KB |
ok, correct split |
112 |
Correct |
125 ms |
37200 KB |
ok, correct split |
113 |
Correct |
109 ms |
37320 KB |
ok, correct split |
114 |
Correct |
15 ms |
21084 KB |
ok, correct split |
115 |
Correct |
16 ms |
20828 KB |
ok, correct split |
116 |
Correct |
117 ms |
36940 KB |
ok, correct split |
117 |
Correct |
103 ms |
36812 KB |
ok, correct split |
118 |
Correct |
117 ms |
34004 KB |
ok, correct split |
119 |
Correct |
93 ms |
33876 KB |
ok, correct split |
120 |
Correct |
101 ms |
39596 KB |
ok, correct split |
121 |
Correct |
84 ms |
33688 KB |
ok, no valid answer |
122 |
Correct |
76 ms |
34132 KB |
ok, no valid answer |
123 |
Correct |
117 ms |
38212 KB |
ok, no valid answer |
124 |
Correct |
200 ms |
37880 KB |
ok, no valid answer |
125 |
Correct |
114 ms |
35788 KB |
ok, no valid answer |
126 |
Correct |
122 ms |
33152 KB |
ok, no valid answer |
127 |
Correct |
114 ms |
36304 KB |
ok, no valid answer |