#include <bits/stdc++.h>
#include "split.h"
using namespace std;
vector <int> v[100000],arb[100000];
vector <pair<int,int>> much[100000];
pair <int,int> vec[3];
pair <int,int> vlu[100001];
int rez[100000],mar[100000],tata[100000],marc2[100000],where[100000],cnt,cat,cnt2,centr;
int coada[100000];
bool marc[100000];
void dfscolor(int nod,int papa,int ind)
{
if(vec[ind].first==0)
return;
rez[nod]=vec[ind].second;
vec[ind].first--;
for(auto it:arb[nod])
if(it!=papa&&it!=centr&&!rez[it])
dfscolor(it,nod,ind);
}
void dfs(int nod,int papa)
{
mar[nod]=1;
tata[nod]=papa;
for(auto it:v[nod])
if(it!=papa&&!mar[it])
{
arb[nod].push_back(it);
arb[it].push_back(nod);
dfs(it,nod);
mar[nod]+=mar[it];
}
}
void dfsnew(int nod,int papa)
{
mar[nod]=1;
for(auto it:arb[nod])
if(it!=papa)
{
dfsnew(it,nod);
mar[nod]+=mar[it];
}
}
int dfscrip(int nod,int papa)
{
int cnt1=1;
where[nod]=cnt;
for(auto it:arb[nod])
if(it!=papa)
cnt1+=dfscrip(it,nod);
return cnt1;
}
void dfsnushce(int nod,int val)
{
if(cat>=vec[0].first)
return;
cat+=vlu[nod].second;
marc2[nod]=val;
if(cat>=vec[0].first)
return;
for(auto it:much[nod])
if(cat<vec[0].first&&!marc2[it.first])
{
coada[++cnt2]=it.second;
dfsnushce(it.first,val);
}
}
vector<int> find_split(int n, int a, int b, int c, vector<int> p, vector<int> q)
{
vec[0]= {a,1};
vec[1]= {b,2};
vec[2]= {c,3};
sort(vec,vec+3);
int i;
for(i=0; i<p.size(); i++)
{
v[p[i]].push_back(q[i]);
v[q[i]].push_back(p[i]);
}
dfs(0,-1);
for(i=0; i<n; i++)
{
int max1=0,s=1;
for(auto it:arb[i])
if(it!=tata[i])
{
max1=max(max1,mar[it]);
s+=mar[it];
}
if(max(max1,n-s)<=n/2)
centr=i;
}
dfsnew(centr,-1);
for(auto it:arb[centr]){
cnt++;
vlu[cnt]= {it,dfscrip(it,centr)};
}
for(i=0; i<n; i++)
if(i!=centr)
for(auto it:v[i])
if(where[it]&&where[i]!=where[it])
much[where[i]].push_back({where[it],it});
/*for(i=1;i<=cnt;i++){
cout<<i<<" ";
for(auto it:much[i])
cout<<it.first<<" ";
cout<<'\n';
}*/
int cnt1=0,ok=0,j;
for(i=1; i<=cnt; i++)
if(!marc2[i])
{
cat=0;
dfsnushce(i,++cnt1);
if(cat>=vec[0].first)
{
//cout<<i<<" "<<cat<<'\n';
dfscolor(vlu[i].first,centr,0);
for(j=1; j<=cnt2; j++)
dfscolor(coada[j],-1,0);
ok=1;
break;
}
}
vector <int> ans;
ans.clear();
if(ok==0)
{
for(i=0; i<n; i++)
ans.push_back(0);
return ans;
}
else
{
dfscolor(centr,-1,1);
for(i=0; i<n; i++)
if(rez[i]==0)
rez[i]=vec[2].second;
for(i=0; i<n; i++)
ans.push_back(rez[i]);
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:75:15: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
75 | for(i=0; i<p.size(); i++)
| ~^~~~~~~~~
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
4 ms |
7252 KB |
ok, correct split |
2 |
Correct |
4 ms |
7380 KB |
ok, correct split |
3 |
Correct |
4 ms |
7340 KB |
ok, correct split |
4 |
Correct |
4 ms |
7380 KB |
ok, correct split |
5 |
Correct |
4 ms |
7380 KB |
ok, correct split |
6 |
Correct |
4 ms |
7380 KB |
ok, correct split |
7 |
Correct |
99 ms |
24788 KB |
ok, correct split |
8 |
Correct |
105 ms |
22472 KB |
ok, correct split |
9 |
Correct |
105 ms |
21868 KB |
ok, correct split |
10 |
Correct |
104 ms |
25184 KB |
ok, correct split |
11 |
Correct |
98 ms |
25148 KB |
ok, correct split |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
4 ms |
7252 KB |
ok, correct split |
2 |
Correct |
4 ms |
7380 KB |
ok, correct split |
3 |
Correct |
4 ms |
7312 KB |
ok, correct split |
4 |
Correct |
114 ms |
22432 KB |
ok, correct split |
5 |
Correct |
100 ms |
17660 KB |
ok, correct split |
6 |
Correct |
96 ms |
25200 KB |
ok, correct split |
7 |
Correct |
94 ms |
22340 KB |
ok, correct split |
8 |
Correct |
134 ms |
19392 KB |
ok, correct split |
9 |
Correct |
90 ms |
17380 KB |
ok, correct split |
10 |
Correct |
70 ms |
19060 KB |
ok, correct split |
11 |
Correct |
69 ms |
19020 KB |
ok, correct split |
12 |
Correct |
76 ms |
18996 KB |
ok, correct split |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
4 ms |
7380 KB |
ok, correct split |
2 |
Correct |
100 ms |
17556 KB |
ok, correct split |
3 |
Correct |
32 ms |
11600 KB |
ok, correct split |
4 |
Correct |
4 ms |
7380 KB |
ok, correct split |
5 |
Correct |
95 ms |
19892 KB |
ok, correct split |
6 |
Correct |
124 ms |
19640 KB |
ok, correct split |
7 |
Correct |
112 ms |
19628 KB |
ok, correct split |
8 |
Correct |
106 ms |
20896 KB |
ok, correct split |
9 |
Correct |
90 ms |
19648 KB |
ok, correct split |
10 |
Correct |
26 ms |
10828 KB |
ok, no valid answer |
11 |
Correct |
38 ms |
12364 KB |
ok, no valid answer |
12 |
Correct |
83 ms |
18144 KB |
ok, no valid answer |
13 |
Correct |
112 ms |
17412 KB |
ok, no valid answer |
14 |
Correct |
65 ms |
19048 KB |
ok, no valid answer |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
4 ms |
7380 KB |
ok, correct split |
2 |
Correct |
4 ms |
7252 KB |
ok, no valid answer |
3 |
Correct |
4 ms |
7380 KB |
ok, correct split |
4 |
Correct |
4 ms |
7380 KB |
ok, correct split |
5 |
Correct |
4 ms |
7380 KB |
ok, correct split |
6 |
Correct |
4 ms |
7252 KB |
ok, correct split |
7 |
Correct |
4 ms |
7380 KB |
ok, correct split |
8 |
Correct |
4 ms |
7380 KB |
ok, correct split |
9 |
Correct |
6 ms |
7636 KB |
ok, correct split |
10 |
Correct |
6 ms |
7636 KB |
ok, correct split |
11 |
Correct |
4 ms |
7380 KB |
ok, correct split |
12 |
Correct |
9 ms |
7764 KB |
ok, correct split |
13 |
Correct |
4 ms |
7380 KB |
ok, correct split |
14 |
Correct |
4 ms |
7380 KB |
ok, correct split |
15 |
Correct |
4 ms |
7380 KB |
ok, correct split |
16 |
Correct |
4 ms |
7380 KB |
ok, correct split |
17 |
Correct |
4 ms |
7380 KB |
ok, correct split |
18 |
Correct |
4 ms |
7380 KB |
ok, correct split |
19 |
Correct |
4 ms |
7380 KB |
ok, correct split |
20 |
Correct |
7 ms |
7440 KB |
ok, correct split |
21 |
Correct |
8 ms |
7636 KB |
ok, correct split |
22 |
Correct |
5 ms |
7636 KB |
ok, correct split |
23 |
Correct |
6 ms |
7636 KB |
ok, correct split |
24 |
Correct |
5 ms |
7636 KB |
ok, correct split |
25 |
Correct |
5 ms |
7636 KB |
ok, correct split |
26 |
Correct |
6 ms |
7764 KB |
ok, correct split |
27 |
Correct |
6 ms |
7876 KB |
ok, correct split |
28 |
Correct |
6 ms |
7812 KB |
ok, correct split |
29 |
Correct |
5 ms |
7380 KB |
ok, correct split |
30 |
Correct |
6 ms |
7764 KB |
ok, correct split |
31 |
Correct |
4 ms |
7360 KB |
ok, correct split |
32 |
Correct |
5 ms |
7380 KB |
ok, correct split |
33 |
Correct |
4 ms |
7380 KB |
ok, correct split |
34 |
Correct |
6 ms |
7620 KB |
ok, correct split |
35 |
Correct |
6 ms |
7616 KB |
ok, correct split |
36 |
Correct |
7 ms |
7636 KB |
ok, correct split |
37 |
Correct |
7 ms |
7744 KB |
ok, correct split |
38 |
Correct |
8 ms |
7824 KB |
ok, correct split |
39 |
Correct |
6 ms |
7764 KB |
ok, correct split |
40 |
Correct |
7 ms |
7764 KB |
ok, correct split |
41 |
Correct |
8 ms |
7492 KB |
ok, correct split |
42 |
Correct |
5 ms |
7488 KB |
ok, correct split |
43 |
Correct |
6 ms |
7764 KB |
ok, correct split |
44 |
Correct |
7 ms |
7624 KB |
ok, correct split |
45 |
Correct |
6 ms |
7752 KB |
ok, correct split |
46 |
Correct |
5 ms |
7636 KB |
ok, correct split |
47 |
Correct |
5 ms |
7636 KB |
ok, no valid answer |
48 |
Correct |
5 ms |
7644 KB |
ok, correct split |
49 |
Correct |
5 ms |
7756 KB |
ok, correct split |
50 |
Correct |
4 ms |
7348 KB |
ok, no valid answer |
51 |
Correct |
4 ms |
7348 KB |
ok, no valid answer |
52 |
Correct |
6 ms |
7644 KB |
ok, no valid answer |
53 |
Correct |
6 ms |
7636 KB |
ok, no valid answer |
54 |
Correct |
5 ms |
7684 KB |
ok, no valid answer |
55 |
Correct |
6 ms |
7620 KB |
ok, no valid answer |
56 |
Correct |
6 ms |
7660 KB |
ok, no valid answer |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
4 ms |
7252 KB |
ok, correct split |
2 |
Correct |
4 ms |
7380 KB |
ok, correct split |
3 |
Correct |
4 ms |
7340 KB |
ok, correct split |
4 |
Correct |
4 ms |
7380 KB |
ok, correct split |
5 |
Correct |
4 ms |
7380 KB |
ok, correct split |
6 |
Correct |
4 ms |
7380 KB |
ok, correct split |
7 |
Correct |
99 ms |
24788 KB |
ok, correct split |
8 |
Correct |
105 ms |
22472 KB |
ok, correct split |
9 |
Correct |
105 ms |
21868 KB |
ok, correct split |
10 |
Correct |
104 ms |
25184 KB |
ok, correct split |
11 |
Correct |
98 ms |
25148 KB |
ok, correct split |
12 |
Correct |
4 ms |
7252 KB |
ok, correct split |
13 |
Correct |
4 ms |
7380 KB |
ok, correct split |
14 |
Correct |
4 ms |
7312 KB |
ok, correct split |
15 |
Correct |
114 ms |
22432 KB |
ok, correct split |
16 |
Correct |
100 ms |
17660 KB |
ok, correct split |
17 |
Correct |
96 ms |
25200 KB |
ok, correct split |
18 |
Correct |
94 ms |
22340 KB |
ok, correct split |
19 |
Correct |
134 ms |
19392 KB |
ok, correct split |
20 |
Correct |
90 ms |
17380 KB |
ok, correct split |
21 |
Correct |
70 ms |
19060 KB |
ok, correct split |
22 |
Correct |
69 ms |
19020 KB |
ok, correct split |
23 |
Correct |
76 ms |
18996 KB |
ok, correct split |
24 |
Correct |
4 ms |
7380 KB |
ok, correct split |
25 |
Correct |
100 ms |
17556 KB |
ok, correct split |
26 |
Correct |
32 ms |
11600 KB |
ok, correct split |
27 |
Correct |
4 ms |
7380 KB |
ok, correct split |
28 |
Correct |
95 ms |
19892 KB |
ok, correct split |
29 |
Correct |
124 ms |
19640 KB |
ok, correct split |
30 |
Correct |
112 ms |
19628 KB |
ok, correct split |
31 |
Correct |
106 ms |
20896 KB |
ok, correct split |
32 |
Correct |
90 ms |
19648 KB |
ok, correct split |
33 |
Correct |
26 ms |
10828 KB |
ok, no valid answer |
34 |
Correct |
38 ms |
12364 KB |
ok, no valid answer |
35 |
Correct |
83 ms |
18144 KB |
ok, no valid answer |
36 |
Correct |
112 ms |
17412 KB |
ok, no valid answer |
37 |
Correct |
65 ms |
19048 KB |
ok, no valid answer |
38 |
Correct |
4 ms |
7380 KB |
ok, correct split |
39 |
Correct |
4 ms |
7252 KB |
ok, no valid answer |
40 |
Correct |
4 ms |
7380 KB |
ok, correct split |
41 |
Correct |
4 ms |
7380 KB |
ok, correct split |
42 |
Correct |
4 ms |
7380 KB |
ok, correct split |
43 |
Correct |
4 ms |
7252 KB |
ok, correct split |
44 |
Correct |
4 ms |
7380 KB |
ok, correct split |
45 |
Correct |
4 ms |
7380 KB |
ok, correct split |
46 |
Correct |
6 ms |
7636 KB |
ok, correct split |
47 |
Correct |
6 ms |
7636 KB |
ok, correct split |
48 |
Correct |
4 ms |
7380 KB |
ok, correct split |
49 |
Correct |
9 ms |
7764 KB |
ok, correct split |
50 |
Correct |
4 ms |
7380 KB |
ok, correct split |
51 |
Correct |
4 ms |
7380 KB |
ok, correct split |
52 |
Correct |
4 ms |
7380 KB |
ok, correct split |
53 |
Correct |
4 ms |
7380 KB |
ok, correct split |
54 |
Correct |
4 ms |
7380 KB |
ok, correct split |
55 |
Correct |
4 ms |
7380 KB |
ok, correct split |
56 |
Correct |
4 ms |
7380 KB |
ok, correct split |
57 |
Correct |
7 ms |
7440 KB |
ok, correct split |
58 |
Correct |
8 ms |
7636 KB |
ok, correct split |
59 |
Correct |
5 ms |
7636 KB |
ok, correct split |
60 |
Correct |
6 ms |
7636 KB |
ok, correct split |
61 |
Correct |
5 ms |
7636 KB |
ok, correct split |
62 |
Correct |
5 ms |
7636 KB |
ok, correct split |
63 |
Correct |
6 ms |
7764 KB |
ok, correct split |
64 |
Correct |
6 ms |
7876 KB |
ok, correct split |
65 |
Correct |
6 ms |
7812 KB |
ok, correct split |
66 |
Correct |
5 ms |
7380 KB |
ok, correct split |
67 |
Correct |
6 ms |
7764 KB |
ok, correct split |
68 |
Correct |
4 ms |
7360 KB |
ok, correct split |
69 |
Correct |
5 ms |
7380 KB |
ok, correct split |
70 |
Correct |
4 ms |
7380 KB |
ok, correct split |
71 |
Correct |
6 ms |
7620 KB |
ok, correct split |
72 |
Correct |
6 ms |
7616 KB |
ok, correct split |
73 |
Correct |
7 ms |
7636 KB |
ok, correct split |
74 |
Correct |
7 ms |
7744 KB |
ok, correct split |
75 |
Correct |
8 ms |
7824 KB |
ok, correct split |
76 |
Correct |
6 ms |
7764 KB |
ok, correct split |
77 |
Correct |
7 ms |
7764 KB |
ok, correct split |
78 |
Correct |
8 ms |
7492 KB |
ok, correct split |
79 |
Correct |
5 ms |
7488 KB |
ok, correct split |
80 |
Correct |
6 ms |
7764 KB |
ok, correct split |
81 |
Correct |
7 ms |
7624 KB |
ok, correct split |
82 |
Correct |
6 ms |
7752 KB |
ok, correct split |
83 |
Correct |
5 ms |
7636 KB |
ok, correct split |
84 |
Correct |
5 ms |
7636 KB |
ok, no valid answer |
85 |
Correct |
5 ms |
7644 KB |
ok, correct split |
86 |
Correct |
5 ms |
7756 KB |
ok, correct split |
87 |
Correct |
4 ms |
7348 KB |
ok, no valid answer |
88 |
Correct |
4 ms |
7348 KB |
ok, no valid answer |
89 |
Correct |
6 ms |
7644 KB |
ok, no valid answer |
90 |
Correct |
6 ms |
7636 KB |
ok, no valid answer |
91 |
Correct |
5 ms |
7684 KB |
ok, no valid answer |
92 |
Correct |
6 ms |
7620 KB |
ok, no valid answer |
93 |
Correct |
6 ms |
7660 KB |
ok, no valid answer |
94 |
Correct |
103 ms |
21660 KB |
ok, correct split |
95 |
Correct |
164 ms |
27592 KB |
ok, correct split |
96 |
Correct |
124 ms |
25708 KB |
ok, correct split |
97 |
Correct |
34 ms |
12008 KB |
ok, correct split |
98 |
Correct |
34 ms |
12108 KB |
ok, correct split |
99 |
Correct |
49 ms |
14124 KB |
ok, correct split |
100 |
Correct |
129 ms |
22220 KB |
ok, correct split |
101 |
Correct |
134 ms |
21132 KB |
ok, correct split |
102 |
Correct |
135 ms |
21184 KB |
ok, correct split |
103 |
Correct |
136 ms |
21188 KB |
ok, correct split |
104 |
Correct |
97 ms |
22504 KB |
ok, correct split |
105 |
Correct |
52 ms |
14700 KB |
ok, correct split |
106 |
Correct |
127 ms |
27832 KB |
ok, correct split |
107 |
Correct |
96 ms |
18756 KB |
ok, correct split |
108 |
Correct |
93 ms |
18744 KB |
ok, correct split |
109 |
Correct |
128 ms |
21656 KB |
ok, correct split |
110 |
Correct |
173 ms |
23872 KB |
ok, correct split |
111 |
Correct |
121 ms |
23628 KB |
ok, correct split |
112 |
Correct |
126 ms |
24124 KB |
ok, correct split |
113 |
Correct |
148 ms |
23936 KB |
ok, correct split |
114 |
Correct |
15 ms |
9300 KB |
ok, correct split |
115 |
Correct |
14 ms |
8984 KB |
ok, correct split |
116 |
Correct |
116 ms |
23860 KB |
ok, correct split |
117 |
Correct |
109 ms |
23312 KB |
ok, correct split |
118 |
Correct |
108 ms |
18604 KB |
ok, correct split |
119 |
Correct |
98 ms |
18600 KB |
ok, correct split |
120 |
Correct |
93 ms |
22692 KB |
ok, correct split |
121 |
Correct |
98 ms |
18336 KB |
ok, no valid answer |
122 |
Correct |
105 ms |
18688 KB |
ok, no valid answer |
123 |
Correct |
160 ms |
21972 KB |
ok, no valid answer |
124 |
Correct |
135 ms |
21968 KB |
ok, no valid answer |
125 |
Correct |
91 ms |
20784 KB |
ok, no valid answer |
126 |
Correct |
55 ms |
18952 KB |
ok, no valid answer |
127 |
Correct |
88 ms |
20752 KB |
ok, no valid answer |