#include "simurgh.h"
#include <bits/stdc++.h>
using namespace std;
const int mxN=500, mxM=mxN*(mxN-1)/2;
int n, tin[mxN], low[mxN], dt=1, b[mxM], bs, st[mxM], sth, t[mxN], p[mxN];
vector<int> eu, ev, adj[mxN], ans;
bool c[mxN];
void dfs1(int u=0, int p=-1) {
tin[u]=low[u]=dt++;
for(int e : adj[u]) {
int v=eu[e]^ev[e]^u;
if(!tin[v]) {
st[sth++]=e;
dfs1(v, u);
low[u]=min(low[v], low[u]);
if(low[v]>=tin[u]) {
do {
b[st[--sth]]=bs;
} while(st[sth]^e);
++bs;
}
} else if(v^p&&tin[v]<tin[u]) {
st[sth++]=e;
low[u]=min(tin[v], low[u]);
}
}
}
int find(int x) {
return x^p[x]?p[x]=find(p[x]):x;
}
bool join(int x, int y) {
if((x=find(x))==(y=find(y)))
return 0;
p[x]=y;
return 1;
}
void fst(vector<int> w, vector<int> &r) {
for(int i=0; i<eu.size(); ++i)
if(w[i]&&join(eu[i], ev[i]))
r.push_back(i);
}
void dc(vector<int> &a, vector<int> &b, vector<int> &c, int l, int r, int s1, int s2) {
if(s2==s1)
return;
if(l==r) {
c.push_back(b[l]);
return;
}
int m=(l+r)/2;
vector<int> d=a;
for(int i=l; i<=m; ++i)
d[i]=b[i];
int sl=count_common_roads(d);
dc(a, b, c, l, m, s1, sl);
dc(a, b, c, m+1, r, s1, s2-sl+s1);
}
void dfs2(int u=0, int pe=-1) {
vector<int> v1, v2, r;
for(int e : adj[u]) {
int v=eu[e]^ev[e]^u;
if(c[v])
continue;
if(t[v]<0) {
v2.push_back(e);
t[v]=e;
} else
v1.push_back(e);
}
if(v1.size()) {
iota(p, p+n, 0);
vector<int> s1;
for(int i=0; i<v1.size(); ++i) {
s1.push_back(t[eu[v1[i]]^ev[v1[i]]^u]);
join(eu[s1.back()], ev[s1.back()]);
}
for(int e : ans) {
s1.push_back(e);
join(eu[e], ev[e]);
}
vector<int> w(eu.size(), 1);
for(int e : v1) {
w[e]=0;
w[t[eu[e]^ev[e]^u]]=0;
}
fst(w, s1);
vector<int> s2=v1;
s2.insert(s2.end(), s1.begin()+v1.size(), s1.end());
dc(s1, v1, r, 0, v1.size()-1, count_common_roads(s1), count_common_roads(s2));
}
if(v2.size()) {
vector<int> w(eu.size(), 1);
for(int e : adj[u])
w[e]=0;
iota(p, p+n, 0);
vector<int> s;
fst(w, s);
for(int e : r)
if(join(eu[e], ev[e]))
s.push_back(e);
if(~pe&&join(eu[pe], ev[pe]))
s.push_back(pe);
for(int e : adj[u])
if(join(eu[e], ev[e]))
s.push_back(e);
int bc=count_common_roads(s);
sort(v2.begin(), v2.end(), [](const int &i, const int &j) {
return b[i]<b[j];
});
for(int i=0, j=0; i<v2.size(); ) {
for(; j<v2.size()&&b[v2[j]]==b[v2[i]]; ++j);
int o;
for(int e : s)
if(b[e]==b[v2[i]]&&(eu[e]==u||ev[e]==u))
o=e;
s.erase(find(s.begin(), s.end(), o));
vector<int> a;
int cm=bc;
for(; i<j; ++i) {
s.push_back(v2[i]);
int cc=count_common_roads(s);
if(cc>cm) {
cm=cc;
a=vector<int>();
}
if(cc==cm)
a.push_back(v2[i]);
s.pop_back();
}
s.push_back(o);
r.insert(r.end(), a.begin(), a.end());
}
}
for(int e : r) {
ans.push_back(e);
c[eu[e]^ev[e]^u]=1;
}
for(int e : r)
dfs2(eu[e]^ev[e]^u, e);
}
vector<int> find_roads(int n, vector<int> u, vector<int> v) {
::n=n;
eu=u;
ev=v;
for(int i=0; i<u.size(); ++i) {
adj[u[i]].push_back(i);
adj[v[i]].push_back(i);
}
dfs1();
c[0]=1;
memset(t, -1, 4*n);
dfs2();
return ans;
}
Compilation message
simurgh.cpp: In function 'void fst(std::vector<int>, std::vector<int>&)':
simurgh.cpp:43:16: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
for(int i=0; i<eu.size(); ++i)
~^~~~~~~~~~
simurgh.cpp: In function 'void dfs2(int, int)':
simurgh.cpp:79:17: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
for(int i=0; i<v1.size(); ++i) {
~^~~~~~~~~~
simurgh.cpp:116:22: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
for(int i=0, j=0; i<v2.size(); ) {
~^~~~~~~~~~
simurgh.cpp:117:11: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
for(; j<v2.size()&&b[v2[j]]==b[v2[i]]; ++j);
~^~~~~~~~~~
simurgh.cpp: In function 'std::vector<int> find_roads(int, std::vector<int>, std::vector<int>)':
simurgh.cpp:152:16: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
for(int i=0; i<u.size(); ++i) {
~^~~~~~~~~
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
2 ms |
376 KB |
correct |
2 |
Correct |
2 ms |
376 KB |
correct |
3 |
Correct |
2 ms |
376 KB |
correct |
4 |
Correct |
2 ms |
376 KB |
correct |
5 |
Correct |
2 ms |
376 KB |
correct |
6 |
Correct |
2 ms |
376 KB |
correct |
7 |
Correct |
2 ms |
376 KB |
correct |
8 |
Correct |
2 ms |
376 KB |
correct |
9 |
Correct |
2 ms |
376 KB |
correct |
10 |
Correct |
2 ms |
376 KB |
correct |
11 |
Correct |
2 ms |
376 KB |
correct |
12 |
Correct |
2 ms |
376 KB |
correct |
13 |
Correct |
2 ms |
376 KB |
correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
2 ms |
376 KB |
correct |
2 |
Correct |
2 ms |
376 KB |
correct |
3 |
Correct |
2 ms |
376 KB |
correct |
4 |
Correct |
2 ms |
376 KB |
correct |
5 |
Correct |
2 ms |
376 KB |
correct |
6 |
Correct |
2 ms |
376 KB |
correct |
7 |
Correct |
2 ms |
376 KB |
correct |
8 |
Correct |
2 ms |
376 KB |
correct |
9 |
Correct |
2 ms |
376 KB |
correct |
10 |
Correct |
2 ms |
376 KB |
correct |
11 |
Correct |
2 ms |
376 KB |
correct |
12 |
Correct |
2 ms |
376 KB |
correct |
13 |
Correct |
2 ms |
376 KB |
correct |
14 |
Correct |
3 ms |
380 KB |
correct |
15 |
Correct |
2 ms |
380 KB |
correct |
16 |
Correct |
3 ms |
376 KB |
correct |
17 |
Correct |
3 ms |
376 KB |
correct |
18 |
Correct |
3 ms |
376 KB |
correct |
19 |
Correct |
3 ms |
376 KB |
correct |
20 |
Correct |
3 ms |
376 KB |
correct |
21 |
Correct |
3 ms |
376 KB |
correct |
22 |
Correct |
3 ms |
380 KB |
correct |
23 |
Correct |
3 ms |
504 KB |
correct |
24 |
Correct |
3 ms |
376 KB |
correct |
25 |
Correct |
2 ms |
376 KB |
correct |
26 |
Correct |
3 ms |
376 KB |
correct |
27 |
Correct |
3 ms |
376 KB |
correct |
28 |
Correct |
3 ms |
376 KB |
correct |
29 |
Correct |
2 ms |
376 KB |
correct |
30 |
Correct |
3 ms |
376 KB |
correct |
31 |
Correct |
3 ms |
376 KB |
correct |
32 |
Correct |
3 ms |
376 KB |
correct |
33 |
Correct |
3 ms |
376 KB |
correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
2 ms |
376 KB |
correct |
2 |
Correct |
2 ms |
376 KB |
correct |
3 |
Correct |
2 ms |
376 KB |
correct |
4 |
Correct |
2 ms |
376 KB |
correct |
5 |
Correct |
2 ms |
376 KB |
correct |
6 |
Correct |
2 ms |
376 KB |
correct |
7 |
Correct |
2 ms |
376 KB |
correct |
8 |
Correct |
2 ms |
376 KB |
correct |
9 |
Correct |
2 ms |
376 KB |
correct |
10 |
Correct |
2 ms |
376 KB |
correct |
11 |
Correct |
2 ms |
376 KB |
correct |
12 |
Correct |
2 ms |
376 KB |
correct |
13 |
Correct |
2 ms |
376 KB |
correct |
14 |
Correct |
3 ms |
380 KB |
correct |
15 |
Correct |
2 ms |
380 KB |
correct |
16 |
Correct |
3 ms |
376 KB |
correct |
17 |
Correct |
3 ms |
376 KB |
correct |
18 |
Correct |
3 ms |
376 KB |
correct |
19 |
Correct |
3 ms |
376 KB |
correct |
20 |
Correct |
3 ms |
376 KB |
correct |
21 |
Correct |
3 ms |
376 KB |
correct |
22 |
Correct |
3 ms |
380 KB |
correct |
23 |
Correct |
3 ms |
504 KB |
correct |
24 |
Correct |
3 ms |
376 KB |
correct |
25 |
Correct |
2 ms |
376 KB |
correct |
26 |
Correct |
3 ms |
376 KB |
correct |
27 |
Correct |
3 ms |
376 KB |
correct |
28 |
Correct |
3 ms |
376 KB |
correct |
29 |
Correct |
2 ms |
376 KB |
correct |
30 |
Correct |
3 ms |
376 KB |
correct |
31 |
Correct |
3 ms |
376 KB |
correct |
32 |
Correct |
3 ms |
376 KB |
correct |
33 |
Correct |
3 ms |
376 KB |
correct |
34 |
Correct |
105 ms |
1824 KB |
correct |
35 |
Correct |
104 ms |
1892 KB |
correct |
36 |
Correct |
74 ms |
1460 KB |
correct |
37 |
Correct |
19 ms |
424 KB |
correct |
38 |
Correct |
111 ms |
1964 KB |
correct |
39 |
Correct |
86 ms |
1672 KB |
correct |
40 |
Correct |
63 ms |
1448 KB |
correct |
41 |
Correct |
14 ms |
1784 KB |
correct |
42 |
Correct |
14 ms |
1916 KB |
correct |
43 |
Correct |
52 ms |
1212 KB |
correct |
44 |
Correct |
45 ms |
1016 KB |
correct |
45 |
Correct |
50 ms |
1104 KB |
correct |
46 |
Correct |
42 ms |
1016 KB |
correct |
47 |
Correct |
25 ms |
632 KB |
correct |
48 |
Correct |
8 ms |
376 KB |
correct |
49 |
Correct |
14 ms |
504 KB |
correct |
50 |
Correct |
25 ms |
636 KB |
correct |
51 |
Correct |
51 ms |
1016 KB |
correct |
52 |
Correct |
45 ms |
1020 KB |
correct |
53 |
Correct |
43 ms |
888 KB |
correct |
54 |
Correct |
53 ms |
1144 KB |
correct |
55 |
Correct |
54 ms |
1144 KB |
correct |
56 |
Correct |
54 ms |
1244 KB |
correct |
57 |
Correct |
54 ms |
1016 KB |
correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
2 ms |
376 KB |
correct |
2 |
Correct |
2 ms |
376 KB |
correct |
3 |
Correct |
482 ms |
4588 KB |
correct |
4 |
Correct |
955 ms |
6580 KB |
correct |
5 |
Correct |
942 ms |
6672 KB |
correct |
6 |
Correct |
57 ms |
6392 KB |
correct |
7 |
Correct |
236 ms |
6464 KB |
correct |
8 |
Correct |
280 ms |
6436 KB |
correct |
9 |
Correct |
916 ms |
6620 KB |
correct |
10 |
Correct |
933 ms |
6928 KB |
correct |
11 |
Correct |
929 ms |
6560 KB |
correct |
12 |
Correct |
949 ms |
7216 KB |
correct |
13 |
Correct |
2 ms |
376 KB |
correct |
14 |
Correct |
528 ms |
6664 KB |
correct |
15 |
Correct |
933 ms |
6916 KB |
correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
2 ms |
376 KB |
correct |
2 |
Correct |
2 ms |
376 KB |
correct |
3 |
Correct |
2 ms |
376 KB |
correct |
4 |
Correct |
2 ms |
376 KB |
correct |
5 |
Correct |
2 ms |
376 KB |
correct |
6 |
Correct |
2 ms |
376 KB |
correct |
7 |
Correct |
2 ms |
376 KB |
correct |
8 |
Correct |
2 ms |
376 KB |
correct |
9 |
Correct |
2 ms |
376 KB |
correct |
10 |
Correct |
2 ms |
376 KB |
correct |
11 |
Correct |
2 ms |
376 KB |
correct |
12 |
Correct |
2 ms |
376 KB |
correct |
13 |
Correct |
2 ms |
376 KB |
correct |
14 |
Correct |
3 ms |
380 KB |
correct |
15 |
Correct |
2 ms |
380 KB |
correct |
16 |
Correct |
3 ms |
376 KB |
correct |
17 |
Correct |
3 ms |
376 KB |
correct |
18 |
Correct |
3 ms |
376 KB |
correct |
19 |
Correct |
3 ms |
376 KB |
correct |
20 |
Correct |
3 ms |
376 KB |
correct |
21 |
Correct |
3 ms |
376 KB |
correct |
22 |
Correct |
3 ms |
380 KB |
correct |
23 |
Correct |
3 ms |
504 KB |
correct |
24 |
Correct |
3 ms |
376 KB |
correct |
25 |
Correct |
2 ms |
376 KB |
correct |
26 |
Correct |
3 ms |
376 KB |
correct |
27 |
Correct |
3 ms |
376 KB |
correct |
28 |
Correct |
3 ms |
376 KB |
correct |
29 |
Correct |
2 ms |
376 KB |
correct |
30 |
Correct |
3 ms |
376 KB |
correct |
31 |
Correct |
3 ms |
376 KB |
correct |
32 |
Correct |
3 ms |
376 KB |
correct |
33 |
Correct |
3 ms |
376 KB |
correct |
34 |
Correct |
105 ms |
1824 KB |
correct |
35 |
Correct |
104 ms |
1892 KB |
correct |
36 |
Correct |
74 ms |
1460 KB |
correct |
37 |
Correct |
19 ms |
424 KB |
correct |
38 |
Correct |
111 ms |
1964 KB |
correct |
39 |
Correct |
86 ms |
1672 KB |
correct |
40 |
Correct |
63 ms |
1448 KB |
correct |
41 |
Correct |
14 ms |
1784 KB |
correct |
42 |
Correct |
14 ms |
1916 KB |
correct |
43 |
Correct |
52 ms |
1212 KB |
correct |
44 |
Correct |
45 ms |
1016 KB |
correct |
45 |
Correct |
50 ms |
1104 KB |
correct |
46 |
Correct |
42 ms |
1016 KB |
correct |
47 |
Correct |
25 ms |
632 KB |
correct |
48 |
Correct |
8 ms |
376 KB |
correct |
49 |
Correct |
14 ms |
504 KB |
correct |
50 |
Correct |
25 ms |
636 KB |
correct |
51 |
Correct |
51 ms |
1016 KB |
correct |
52 |
Correct |
45 ms |
1020 KB |
correct |
53 |
Correct |
43 ms |
888 KB |
correct |
54 |
Correct |
53 ms |
1144 KB |
correct |
55 |
Correct |
54 ms |
1144 KB |
correct |
56 |
Correct |
54 ms |
1244 KB |
correct |
57 |
Correct |
54 ms |
1016 KB |
correct |
58 |
Correct |
2 ms |
376 KB |
correct |
59 |
Correct |
2 ms |
376 KB |
correct |
60 |
Correct |
482 ms |
4588 KB |
correct |
61 |
Correct |
955 ms |
6580 KB |
correct |
62 |
Correct |
942 ms |
6672 KB |
correct |
63 |
Correct |
57 ms |
6392 KB |
correct |
64 |
Correct |
236 ms |
6464 KB |
correct |
65 |
Correct |
280 ms |
6436 KB |
correct |
66 |
Correct |
916 ms |
6620 KB |
correct |
67 |
Correct |
933 ms |
6928 KB |
correct |
68 |
Correct |
929 ms |
6560 KB |
correct |
69 |
Correct |
949 ms |
7216 KB |
correct |
70 |
Correct |
2 ms |
376 KB |
correct |
71 |
Correct |
528 ms |
6664 KB |
correct |
72 |
Correct |
933 ms |
6916 KB |
correct |
73 |
Correct |
2 ms |
376 KB |
correct |
74 |
Correct |
933 ms |
6588 KB |
correct |
75 |
Correct |
910 ms |
6548 KB |
correct |
76 |
Correct |
298 ms |
2644 KB |
correct |
77 |
Correct |
948 ms |
6784 KB |
correct |
78 |
Correct |
945 ms |
6644 KB |
correct |
79 |
Correct |
961 ms |
6728 KB |
correct |
80 |
Correct |
920 ms |
6644 KB |
correct |
81 |
Correct |
57 ms |
5624 KB |
correct |
82 |
Correct |
915 ms |
6984 KB |
correct |
83 |
Correct |
516 ms |
3696 KB |
correct |
84 |
Correct |
546 ms |
4424 KB |
correct |
85 |
Correct |
497 ms |
3912 KB |
correct |
86 |
Correct |
350 ms |
2820 KB |
correct |
87 |
Correct |
255 ms |
2224 KB |
correct |
88 |
Correct |
213 ms |
1852 KB |
correct |
89 |
Correct |
189 ms |
1768 KB |
correct |
90 |
Correct |
164 ms |
1460 KB |
correct |
91 |
Correct |
52 ms |
504 KB |
correct |
92 |
Correct |
30 ms |
376 KB |
correct |
93 |
Correct |
493 ms |
4240 KB |
correct |
94 |
Correct |
375 ms |
2748 KB |
correct |
95 |
Correct |
330 ms |
2756 KB |
correct |
96 |
Correct |
181 ms |
1592 KB |
correct |
97 |
Correct |
219 ms |
1792 KB |
correct |
98 |
Correct |
253 ms |
2268 KB |
correct |
99 |
Correct |
225 ms |
1816 KB |
correct |
100 |
Correct |
76 ms |
760 KB |
correct |
101 |
Correct |
29 ms |
508 KB |
correct |
102 |
Correct |
501 ms |
3580 KB |
correct |
103 |
Correct |
532 ms |
3484 KB |
correct |
104 |
Correct |
505 ms |
3744 KB |
correct |
105 |
Correct |
486 ms |
3672 KB |
correct |