#include<bits/stdc++.h>
using namespace std;
using I=int;
using B=bool;
const I N=100000;
const I LOGN=17;
vector<pair<I,I>>adjs[N];
set<I>incs[N];
vector<I>rems[N];
I ancs[N][LOGN];
I tbgs[N],teds[N];
vector<I>ress;
I t=0;
I k;
void dfs1(I a,I p){
tbgs[a]=t++;
ancs[a][0]=p;
for(I i=1;i<LOGN;i++)ancs[a][i]=ancs[ancs[a][i-1]][i-1];
for(auto[b,i]:adjs[a])if(b!=p)dfs1(b,a);
teds[a]=t;
}
B anc(I a,I b){
return tbgs[a]<=tbgs[b]&&teds[a]>=teds[b];
}
I lca(I a,I b){
if(anc(a,b))return a;
if(anc(b,a))return b;
for(I i=LOGN-1;i>=0;i--)if(!anc(ancs[a][i],b))a=ancs[a][i];
return ancs[a][0];
}
void dfs2(I a,I p=-1){
for(auto[b,i]:adjs[a])if(b!=p){
dfs2(b,a);
if(incs[b].size()>=k)ress.push_back(i);
if(incs[a].size()<incs[b].size())swap(incs[a],incs[b]);
for(auto x:incs[b])incs[a].insert(x);
}
for(auto x:rems[a])incs[a].erase(x);
}
I main(){
cin.tie(0)->sync_with_stdio(0);
I n,m;cin>>n>>m>>k;
for(I i=0;i<n-1;i++){
I a,b;cin>>a>>b,a--,b--;
adjs[a].push_back({b,i});
adjs[b].push_back({a,i});
}
dfs1(0,0);
for(I i=0;i<m;i++){
I s;cin>>s;
I y=-1;
for(I j=0;j<s;j++){
I x;cin>>x,x--;
y=y==-1?x:lca(y,x);
incs[x].insert(i);
}
rems[y].push_back(i);
}
dfs2(0);
sort(ress.begin(),ress.end());
printf("%i\n",ress.size());
for(auto x:ress)printf("%i ",x+1);
}
Compilation message
railway.cpp: In function 'void dfs2(I, I)':
railway.cpp:34:22: warning: comparison of integer expressions of different signedness: 'std::set<int>::size_type' {aka 'long unsigned int'} and 'I' {aka 'int'} [-Wsign-compare]
34 | if(incs[b].size()>=k)ress.push_back(i);
| ~~~~~~~~~~~~~~^~~
railway.cpp: In function 'I main()':
railway.cpp:61:12: warning: format '%i' expects argument of type 'int', but argument 2 has type 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wformat=]
61 | printf("%i\n",ress.size());
| ~^ ~~~~~~~~~~~
| | |
| int std::vector<int>::size_type {aka long unsigned int}
| %li
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
5 ms |
9728 KB |
Output is correct |
2 |
Correct |
9 ms |
11020 KB |
Output is correct |
3 |
Correct |
12 ms |
10964 KB |
Output is correct |
4 |
Correct |
5 ms |
9684 KB |
Output is correct |
5 |
Correct |
5 ms |
9720 KB |
Output is correct |
6 |
Correct |
10 ms |
11772 KB |
Output is correct |
7 |
Correct |
9 ms |
11092 KB |
Output is correct |
8 |
Correct |
9 ms |
11016 KB |
Output is correct |
9 |
Correct |
9 ms |
11220 KB |
Output is correct |
10 |
Correct |
5 ms |
9684 KB |
Output is correct |
11 |
Correct |
5 ms |
9684 KB |
Output is correct |
12 |
Correct |
5 ms |
9728 KB |
Output is correct |
13 |
Correct |
5 ms |
9692 KB |
Output is correct |
14 |
Correct |
5 ms |
9704 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
5 ms |
9728 KB |
Output is correct |
2 |
Correct |
9 ms |
11020 KB |
Output is correct |
3 |
Correct |
12 ms |
10964 KB |
Output is correct |
4 |
Correct |
5 ms |
9684 KB |
Output is correct |
5 |
Correct |
5 ms |
9720 KB |
Output is correct |
6 |
Correct |
10 ms |
11772 KB |
Output is correct |
7 |
Correct |
9 ms |
11092 KB |
Output is correct |
8 |
Correct |
9 ms |
11016 KB |
Output is correct |
9 |
Correct |
9 ms |
11220 KB |
Output is correct |
10 |
Correct |
5 ms |
9684 KB |
Output is correct |
11 |
Correct |
5 ms |
9684 KB |
Output is correct |
12 |
Correct |
5 ms |
9728 KB |
Output is correct |
13 |
Correct |
5 ms |
9692 KB |
Output is correct |
14 |
Correct |
5 ms |
9704 KB |
Output is correct |
15 |
Correct |
37 ms |
16716 KB |
Output is correct |
16 |
Correct |
38 ms |
16900 KB |
Output is correct |
17 |
Correct |
38 ms |
16892 KB |
Output is correct |
18 |
Correct |
11 ms |
11988 KB |
Output is correct |
19 |
Correct |
10 ms |
11272 KB |
Output is correct |
20 |
Correct |
39 ms |
16212 KB |
Output is correct |
21 |
Correct |
42 ms |
16156 KB |
Output is correct |
22 |
Correct |
5 ms |
9684 KB |
Output is correct |
23 |
Correct |
10 ms |
11020 KB |
Output is correct |
24 |
Correct |
10 ms |
11016 KB |
Output is correct |
25 |
Correct |
5 ms |
9684 KB |
Output is correct |
26 |
Correct |
5 ms |
9684 KB |
Output is correct |
27 |
Correct |
10 ms |
11888 KB |
Output is correct |
28 |
Correct |
10 ms |
11128 KB |
Output is correct |
29 |
Correct |
12 ms |
11092 KB |
Output is correct |
30 |
Correct |
13 ms |
11220 KB |
Output is correct |
31 |
Correct |
5 ms |
9684 KB |
Output is correct |
32 |
Correct |
6 ms |
9628 KB |
Output is correct |
33 |
Correct |
5 ms |
9684 KB |
Output is correct |
34 |
Correct |
5 ms |
9684 KB |
Output is correct |
35 |
Correct |
5 ms |
9684 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
127 ms |
45536 KB |
Output is correct |
2 |
Correct |
5 ms |
9684 KB |
Output is correct |
3 |
Correct |
131 ms |
45156 KB |
Output is correct |
4 |
Correct |
114 ms |
44876 KB |
Output is correct |
5 |
Correct |
95 ms |
43564 KB |
Output is correct |
6 |
Correct |
124 ms |
46028 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
118 ms |
37840 KB |
Output is correct |
2 |
Correct |
161 ms |
32748 KB |
Output is correct |
3 |
Correct |
176 ms |
38268 KB |
Output is correct |
4 |
Correct |
161 ms |
38136 KB |
Output is correct |
5 |
Correct |
191 ms |
38452 KB |
Output is correct |
6 |
Correct |
121 ms |
38152 KB |
Output is correct |
7 |
Correct |
126 ms |
38092 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
118 ms |
37840 KB |
Output is correct |
2 |
Correct |
161 ms |
32748 KB |
Output is correct |
3 |
Correct |
176 ms |
38268 KB |
Output is correct |
4 |
Correct |
161 ms |
38136 KB |
Output is correct |
5 |
Correct |
191 ms |
38452 KB |
Output is correct |
6 |
Correct |
121 ms |
38152 KB |
Output is correct |
7 |
Correct |
126 ms |
38092 KB |
Output is correct |
8 |
Correct |
110 ms |
36724 KB |
Output is correct |
9 |
Correct |
113 ms |
36660 KB |
Output is correct |
10 |
Correct |
85 ms |
43452 KB |
Output is correct |
11 |
Correct |
87 ms |
43412 KB |
Output is correct |
12 |
Correct |
72 ms |
27168 KB |
Output is correct |
13 |
Correct |
73 ms |
27092 KB |
Output is correct |
14 |
Correct |
132 ms |
33152 KB |
Output is correct |
15 |
Correct |
134 ms |
33356 KB |
Output is correct |
16 |
Correct |
181 ms |
39292 KB |
Output is correct |
17 |
Correct |
164 ms |
39628 KB |
Output is correct |
18 |
Correct |
161 ms |
37644 KB |
Output is correct |
19 |
Correct |
154 ms |
33404 KB |
Output is correct |
20 |
Correct |
120 ms |
38308 KB |
Output is correct |
21 |
Correct |
123 ms |
38340 KB |
Output is correct |
22 |
Correct |
126 ms |
38400 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
5 ms |
9728 KB |
Output is correct |
2 |
Correct |
9 ms |
11020 KB |
Output is correct |
3 |
Correct |
12 ms |
10964 KB |
Output is correct |
4 |
Correct |
5 ms |
9684 KB |
Output is correct |
5 |
Correct |
5 ms |
9720 KB |
Output is correct |
6 |
Correct |
10 ms |
11772 KB |
Output is correct |
7 |
Correct |
9 ms |
11092 KB |
Output is correct |
8 |
Correct |
9 ms |
11016 KB |
Output is correct |
9 |
Correct |
9 ms |
11220 KB |
Output is correct |
10 |
Correct |
5 ms |
9684 KB |
Output is correct |
11 |
Correct |
5 ms |
9684 KB |
Output is correct |
12 |
Correct |
5 ms |
9728 KB |
Output is correct |
13 |
Correct |
5 ms |
9692 KB |
Output is correct |
14 |
Correct |
5 ms |
9704 KB |
Output is correct |
15 |
Correct |
37 ms |
16716 KB |
Output is correct |
16 |
Correct |
38 ms |
16900 KB |
Output is correct |
17 |
Correct |
38 ms |
16892 KB |
Output is correct |
18 |
Correct |
11 ms |
11988 KB |
Output is correct |
19 |
Correct |
10 ms |
11272 KB |
Output is correct |
20 |
Correct |
39 ms |
16212 KB |
Output is correct |
21 |
Correct |
42 ms |
16156 KB |
Output is correct |
22 |
Correct |
5 ms |
9684 KB |
Output is correct |
23 |
Correct |
10 ms |
11020 KB |
Output is correct |
24 |
Correct |
10 ms |
11016 KB |
Output is correct |
25 |
Correct |
5 ms |
9684 KB |
Output is correct |
26 |
Correct |
5 ms |
9684 KB |
Output is correct |
27 |
Correct |
10 ms |
11888 KB |
Output is correct |
28 |
Correct |
10 ms |
11128 KB |
Output is correct |
29 |
Correct |
12 ms |
11092 KB |
Output is correct |
30 |
Correct |
13 ms |
11220 KB |
Output is correct |
31 |
Correct |
5 ms |
9684 KB |
Output is correct |
32 |
Correct |
6 ms |
9628 KB |
Output is correct |
33 |
Correct |
5 ms |
9684 KB |
Output is correct |
34 |
Correct |
5 ms |
9684 KB |
Output is correct |
35 |
Correct |
5 ms |
9684 KB |
Output is correct |
36 |
Correct |
127 ms |
45536 KB |
Output is correct |
37 |
Correct |
5 ms |
9684 KB |
Output is correct |
38 |
Correct |
131 ms |
45156 KB |
Output is correct |
39 |
Correct |
114 ms |
44876 KB |
Output is correct |
40 |
Correct |
95 ms |
43564 KB |
Output is correct |
41 |
Correct |
124 ms |
46028 KB |
Output is correct |
42 |
Correct |
118 ms |
37840 KB |
Output is correct |
43 |
Correct |
161 ms |
32748 KB |
Output is correct |
44 |
Correct |
176 ms |
38268 KB |
Output is correct |
45 |
Correct |
161 ms |
38136 KB |
Output is correct |
46 |
Correct |
191 ms |
38452 KB |
Output is correct |
47 |
Correct |
121 ms |
38152 KB |
Output is correct |
48 |
Correct |
126 ms |
38092 KB |
Output is correct |
49 |
Correct |
110 ms |
36724 KB |
Output is correct |
50 |
Correct |
113 ms |
36660 KB |
Output is correct |
51 |
Correct |
85 ms |
43452 KB |
Output is correct |
52 |
Correct |
87 ms |
43412 KB |
Output is correct |
53 |
Correct |
72 ms |
27168 KB |
Output is correct |
54 |
Correct |
73 ms |
27092 KB |
Output is correct |
55 |
Correct |
132 ms |
33152 KB |
Output is correct |
56 |
Correct |
134 ms |
33356 KB |
Output is correct |
57 |
Correct |
181 ms |
39292 KB |
Output is correct |
58 |
Correct |
164 ms |
39628 KB |
Output is correct |
59 |
Correct |
161 ms |
37644 KB |
Output is correct |
60 |
Correct |
154 ms |
33404 KB |
Output is correct |
61 |
Correct |
120 ms |
38308 KB |
Output is correct |
62 |
Correct |
123 ms |
38340 KB |
Output is correct |
63 |
Correct |
126 ms |
38400 KB |
Output is correct |
64 |
Correct |
118 ms |
37296 KB |
Output is correct |
65 |
Correct |
190 ms |
44860 KB |
Output is correct |
66 |
Correct |
157 ms |
40472 KB |
Output is correct |
67 |
Correct |
137 ms |
36336 KB |
Output is correct |
68 |
Correct |
68 ms |
25612 KB |
Output is correct |
69 |
Correct |
71 ms |
25612 KB |
Output is correct |
70 |
Correct |
120 ms |
38536 KB |
Output is correct |
71 |
Correct |
100 ms |
36712 KB |
Output is correct |
72 |
Correct |
5 ms |
9684 KB |
Output is correct |
73 |
Correct |
9 ms |
11012 KB |
Output is correct |
74 |
Correct |
9 ms |
11092 KB |
Output is correct |
75 |
Correct |
5 ms |
9724 KB |
Output is correct |
76 |
Correct |
5 ms |
9684 KB |
Output is correct |
77 |
Correct |
10 ms |
11860 KB |
Output is correct |
78 |
Correct |
9 ms |
11092 KB |
Output is correct |
79 |
Correct |
8 ms |
11092 KB |
Output is correct |
80 |
Correct |
9 ms |
11140 KB |
Output is correct |
81 |
Correct |
5 ms |
9684 KB |
Output is correct |
82 |
Correct |
5 ms |
9684 KB |
Output is correct |
83 |
Correct |
5 ms |
9700 KB |
Output is correct |
84 |
Correct |
5 ms |
9728 KB |
Output is correct |
85 |
Correct |
5 ms |
9684 KB |
Output is correct |
86 |
Correct |
39 ms |
16876 KB |
Output is correct |
87 |
Correct |
40 ms |
16844 KB |
Output is correct |
88 |
Correct |
40 ms |
16844 KB |
Output is correct |
89 |
Correct |
11 ms |
12040 KB |
Output is correct |
90 |
Correct |
10 ms |
11240 KB |
Output is correct |
91 |
Correct |
37 ms |
16212 KB |
Output is correct |
92 |
Correct |
36 ms |
16092 KB |
Output is correct |
93 |
Correct |
6 ms |
9684 KB |
Output is correct |
94 |
Correct |
121 ms |
45544 KB |
Output is correct |
95 |
Correct |
136 ms |
45236 KB |
Output is correct |
96 |
Correct |
117 ms |
44676 KB |
Output is correct |
97 |
Correct |
95 ms |
43608 KB |
Output is correct |
98 |
Correct |
120 ms |
46068 KB |
Output is correct |
99 |
Correct |
179 ms |
41856 KB |
Output is correct |
100 |
Correct |
184 ms |
39096 KB |
Output is correct |
101 |
Correct |
166 ms |
39496 KB |
Output is correct |
102 |
Correct |
149 ms |
33080 KB |
Output is correct |
103 |
Correct |
118 ms |
38092 KB |
Output is correct |
104 |
Correct |
121 ms |
38380 KB |
Output is correct |
105 |
Correct |
123 ms |
38076 KB |
Output is correct |
106 |
Correct |
114 ms |
36836 KB |
Output is correct |
107 |
Correct |
114 ms |
36912 KB |
Output is correct |
108 |
Correct |
82 ms |
43440 KB |
Output is correct |
109 |
Correct |
82 ms |
43400 KB |
Output is correct |
110 |
Correct |
75 ms |
27128 KB |
Output is correct |
111 |
Correct |
75 ms |
27176 KB |
Output is correct |
112 |
Correct |
155 ms |
33056 KB |
Output is correct |
113 |
Correct |
135 ms |
32964 KB |
Output is correct |