#include <bits/stdc++.h>
using namespace std;
const int N = 1e5 + 1;
int n,m,k;
vector<int> adj[N],rail[N];
int am[N],sz[N],big[N],cnt[N],pa[N];
int now,ans[N];
map<pair<int,int>,int> id;
vector<int> an;
void pre(int u,int p){ sz[u] = 1,pa[u] = p; for(int v : adj[u]) if(v!=p) pre(v,u),sz[u]+=sz[v]; }
void add(int u,int p,int k)
{
for(int x : rail[u])
{
if(!cnt[x] or cnt[x]==am[x]) now++;
cnt[x]+=k;
if(!cnt[x] or cnt[x]==am[x]) now--;
}
for(int v : adj[u]) if(v!=p and !big[v]) add(v,u,k);
}
void dfs(int u,int p,int k)
{
int bc = -1,mx = -1;
for(int v : adj[u]) if(v!=p) if(sz[v]>mx) mx = sz[v],bc = v;
for(int v : adj[u]) if(v!=p and v!=bc) dfs(v,u,0);
if(bc!=-1) dfs(bc,u,1),big[bc] = 1;
add(u,p,1);
ans[u] = now;
if(bc!=-1) big[bc] = 0;
if(!k) add(u,p,-1);
}
int main()
{
ios_base::sync_with_stdio(0); cin.tie(0);
cin >> n >> m >> k;
for(int i = 0;i < n-1;i++)
{
int a,b;
cin >> a >> b;
adj[a].push_back(b);
adj[b].push_back(a);
id[{a,b}] = id[{b,a}] = i+1;
}
for(int i = 1;i <= m;i++)
{
cin >> am[i];
for(int j = 0;j < am[i];j++)
{
int x;
cin >> x;
rail[x].push_back(i);
}
}
pre(1,0);
dfs(1,0,1);
for(int i = 2;i <= n;i++)
{
if(ans[i]>=k) an.push_back(id[{i,pa[i]}]);
}
sort(an.begin(),an.end());
cout << an.size() << '\n';
for(int x : an) cout << x << ' ';
}
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
3 ms |
5120 KB |
Output is correct |
2 |
Correct |
15 ms |
6912 KB |
Output is correct |
3 |
Correct |
18 ms |
6912 KB |
Output is correct |
4 |
Correct |
4 ms |
5120 KB |
Output is correct |
5 |
Correct |
5 ms |
5120 KB |
Output is correct |
6 |
Correct |
14 ms |
7424 KB |
Output is correct |
7 |
Correct |
14 ms |
7040 KB |
Output is correct |
8 |
Correct |
12 ms |
6912 KB |
Output is correct |
9 |
Correct |
13 ms |
6912 KB |
Output is correct |
10 |
Correct |
3 ms |
5120 KB |
Output is correct |
11 |
Correct |
3 ms |
5120 KB |
Output is correct |
12 |
Correct |
3 ms |
5120 KB |
Output is correct |
13 |
Correct |
4 ms |
5120 KB |
Output is correct |
14 |
Correct |
4 ms |
5120 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
3 ms |
5120 KB |
Output is correct |
2 |
Correct |
15 ms |
6912 KB |
Output is correct |
3 |
Correct |
18 ms |
6912 KB |
Output is correct |
4 |
Correct |
4 ms |
5120 KB |
Output is correct |
5 |
Correct |
5 ms |
5120 KB |
Output is correct |
6 |
Correct |
14 ms |
7424 KB |
Output is correct |
7 |
Correct |
14 ms |
7040 KB |
Output is correct |
8 |
Correct |
12 ms |
6912 KB |
Output is correct |
9 |
Correct |
13 ms |
6912 KB |
Output is correct |
10 |
Correct |
3 ms |
5120 KB |
Output is correct |
11 |
Correct |
3 ms |
5120 KB |
Output is correct |
12 |
Correct |
3 ms |
5120 KB |
Output is correct |
13 |
Correct |
4 ms |
5120 KB |
Output is correct |
14 |
Correct |
4 ms |
5120 KB |
Output is correct |
15 |
Correct |
31 ms |
7868 KB |
Output is correct |
16 |
Correct |
30 ms |
7928 KB |
Output is correct |
17 |
Correct |
31 ms |
7936 KB |
Output is correct |
18 |
Correct |
14 ms |
7424 KB |
Output is correct |
19 |
Correct |
14 ms |
7040 KB |
Output is correct |
20 |
Correct |
27 ms |
8320 KB |
Output is correct |
21 |
Correct |
26 ms |
8192 KB |
Output is correct |
22 |
Correct |
3 ms |
5120 KB |
Output is correct |
23 |
Correct |
15 ms |
7040 KB |
Output is correct |
24 |
Correct |
15 ms |
6912 KB |
Output is correct |
25 |
Correct |
4 ms |
5120 KB |
Output is correct |
26 |
Correct |
3 ms |
5120 KB |
Output is correct |
27 |
Correct |
14 ms |
7424 KB |
Output is correct |
28 |
Correct |
14 ms |
7168 KB |
Output is correct |
29 |
Correct |
16 ms |
6912 KB |
Output is correct |
30 |
Correct |
13 ms |
6912 KB |
Output is correct |
31 |
Correct |
3 ms |
5120 KB |
Output is correct |
32 |
Correct |
3 ms |
5120 KB |
Output is correct |
33 |
Correct |
3 ms |
5120 KB |
Output is correct |
34 |
Correct |
3 ms |
5120 KB |
Output is correct |
35 |
Correct |
4 ms |
5120 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
144 ms |
35328 KB |
Output is correct |
2 |
Correct |
4 ms |
5120 KB |
Output is correct |
3 |
Correct |
141 ms |
34864 KB |
Output is correct |
4 |
Correct |
156 ms |
34296 KB |
Output is correct |
5 |
Correct |
144 ms |
34896 KB |
Output is correct |
6 |
Correct |
145 ms |
35064 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
175 ms |
29048 KB |
Output is correct |
2 |
Correct |
262 ms |
25536 KB |
Output is correct |
3 |
Correct |
282 ms |
25336 KB |
Output is correct |
4 |
Correct |
281 ms |
25228 KB |
Output is correct |
5 |
Correct |
299 ms |
25336 KB |
Output is correct |
6 |
Correct |
219 ms |
29304 KB |
Output is correct |
7 |
Correct |
211 ms |
29176 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
175 ms |
29048 KB |
Output is correct |
2 |
Correct |
262 ms |
25536 KB |
Output is correct |
3 |
Correct |
282 ms |
25336 KB |
Output is correct |
4 |
Correct |
281 ms |
25228 KB |
Output is correct |
5 |
Correct |
299 ms |
25336 KB |
Output is correct |
6 |
Correct |
219 ms |
29304 KB |
Output is correct |
7 |
Correct |
211 ms |
29176 KB |
Output is correct |
8 |
Correct |
201 ms |
29944 KB |
Output is correct |
9 |
Correct |
219 ms |
29920 KB |
Output is correct |
10 |
Correct |
143 ms |
34804 KB |
Output is correct |
11 |
Correct |
156 ms |
34788 KB |
Output is correct |
12 |
Correct |
166 ms |
25720 KB |
Output is correct |
13 |
Correct |
159 ms |
25592 KB |
Output is correct |
14 |
Correct |
252 ms |
24824 KB |
Output is correct |
15 |
Correct |
293 ms |
24824 KB |
Output is correct |
16 |
Correct |
257 ms |
25208 KB |
Output is correct |
17 |
Correct |
264 ms |
25208 KB |
Output is correct |
18 |
Correct |
275 ms |
25208 KB |
Output is correct |
19 |
Correct |
233 ms |
25592 KB |
Output is correct |
20 |
Correct |
177 ms |
29432 KB |
Output is correct |
21 |
Correct |
221 ms |
29564 KB |
Output is correct |
22 |
Correct |
191 ms |
29432 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
3 ms |
5120 KB |
Output is correct |
2 |
Correct |
15 ms |
6912 KB |
Output is correct |
3 |
Correct |
18 ms |
6912 KB |
Output is correct |
4 |
Correct |
4 ms |
5120 KB |
Output is correct |
5 |
Correct |
5 ms |
5120 KB |
Output is correct |
6 |
Correct |
14 ms |
7424 KB |
Output is correct |
7 |
Correct |
14 ms |
7040 KB |
Output is correct |
8 |
Correct |
12 ms |
6912 KB |
Output is correct |
9 |
Correct |
13 ms |
6912 KB |
Output is correct |
10 |
Correct |
3 ms |
5120 KB |
Output is correct |
11 |
Correct |
3 ms |
5120 KB |
Output is correct |
12 |
Correct |
3 ms |
5120 KB |
Output is correct |
13 |
Correct |
4 ms |
5120 KB |
Output is correct |
14 |
Correct |
4 ms |
5120 KB |
Output is correct |
15 |
Correct |
31 ms |
7868 KB |
Output is correct |
16 |
Correct |
30 ms |
7928 KB |
Output is correct |
17 |
Correct |
31 ms |
7936 KB |
Output is correct |
18 |
Correct |
14 ms |
7424 KB |
Output is correct |
19 |
Correct |
14 ms |
7040 KB |
Output is correct |
20 |
Correct |
27 ms |
8320 KB |
Output is correct |
21 |
Correct |
26 ms |
8192 KB |
Output is correct |
22 |
Correct |
3 ms |
5120 KB |
Output is correct |
23 |
Correct |
15 ms |
7040 KB |
Output is correct |
24 |
Correct |
15 ms |
6912 KB |
Output is correct |
25 |
Correct |
4 ms |
5120 KB |
Output is correct |
26 |
Correct |
3 ms |
5120 KB |
Output is correct |
27 |
Correct |
14 ms |
7424 KB |
Output is correct |
28 |
Correct |
14 ms |
7168 KB |
Output is correct |
29 |
Correct |
16 ms |
6912 KB |
Output is correct |
30 |
Correct |
13 ms |
6912 KB |
Output is correct |
31 |
Correct |
3 ms |
5120 KB |
Output is correct |
32 |
Correct |
3 ms |
5120 KB |
Output is correct |
33 |
Correct |
3 ms |
5120 KB |
Output is correct |
34 |
Correct |
3 ms |
5120 KB |
Output is correct |
35 |
Correct |
4 ms |
5120 KB |
Output is correct |
36 |
Correct |
144 ms |
35328 KB |
Output is correct |
37 |
Correct |
4 ms |
5120 KB |
Output is correct |
38 |
Correct |
141 ms |
34864 KB |
Output is correct |
39 |
Correct |
156 ms |
34296 KB |
Output is correct |
40 |
Correct |
144 ms |
34896 KB |
Output is correct |
41 |
Correct |
145 ms |
35064 KB |
Output is correct |
42 |
Correct |
175 ms |
29048 KB |
Output is correct |
43 |
Correct |
262 ms |
25536 KB |
Output is correct |
44 |
Correct |
282 ms |
25336 KB |
Output is correct |
45 |
Correct |
281 ms |
25228 KB |
Output is correct |
46 |
Correct |
299 ms |
25336 KB |
Output is correct |
47 |
Correct |
219 ms |
29304 KB |
Output is correct |
48 |
Correct |
211 ms |
29176 KB |
Output is correct |
49 |
Correct |
201 ms |
29944 KB |
Output is correct |
50 |
Correct |
219 ms |
29920 KB |
Output is correct |
51 |
Correct |
143 ms |
34804 KB |
Output is correct |
52 |
Correct |
156 ms |
34788 KB |
Output is correct |
53 |
Correct |
166 ms |
25720 KB |
Output is correct |
54 |
Correct |
159 ms |
25592 KB |
Output is correct |
55 |
Correct |
252 ms |
24824 KB |
Output is correct |
56 |
Correct |
293 ms |
24824 KB |
Output is correct |
57 |
Correct |
257 ms |
25208 KB |
Output is correct |
58 |
Correct |
264 ms |
25208 KB |
Output is correct |
59 |
Correct |
275 ms |
25208 KB |
Output is correct |
60 |
Correct |
233 ms |
25592 KB |
Output is correct |
61 |
Correct |
177 ms |
29432 KB |
Output is correct |
62 |
Correct |
221 ms |
29564 KB |
Output is correct |
63 |
Correct |
191 ms |
29432 KB |
Output is correct |
64 |
Correct |
252 ms |
29944 KB |
Output is correct |
65 |
Correct |
363 ms |
26744 KB |
Output is correct |
66 |
Correct |
324 ms |
26152 KB |
Output is correct |
67 |
Correct |
252 ms |
25720 KB |
Output is correct |
68 |
Correct |
178 ms |
25080 KB |
Output is correct |
69 |
Correct |
174 ms |
24952 KB |
Output is correct |
70 |
Correct |
213 ms |
30344 KB |
Output is correct |
71 |
Correct |
194 ms |
30072 KB |
Output is correct |
72 |
Correct |
4 ms |
5120 KB |
Output is correct |
73 |
Correct |
15 ms |
6912 KB |
Output is correct |
74 |
Correct |
15 ms |
6912 KB |
Output is correct |
75 |
Correct |
4 ms |
5120 KB |
Output is correct |
76 |
Correct |
4 ms |
5120 KB |
Output is correct |
77 |
Correct |
14 ms |
7552 KB |
Output is correct |
78 |
Correct |
16 ms |
7016 KB |
Output is correct |
79 |
Correct |
16 ms |
6912 KB |
Output is correct |
80 |
Correct |
13 ms |
6912 KB |
Output is correct |
81 |
Correct |
4 ms |
5120 KB |
Output is correct |
82 |
Correct |
4 ms |
5120 KB |
Output is correct |
83 |
Correct |
4 ms |
5120 KB |
Output is correct |
84 |
Correct |
3 ms |
5120 KB |
Output is correct |
85 |
Correct |
4 ms |
5120 KB |
Output is correct |
86 |
Correct |
31 ms |
7928 KB |
Output is correct |
87 |
Correct |
37 ms |
7936 KB |
Output is correct |
88 |
Correct |
35 ms |
7952 KB |
Output is correct |
89 |
Correct |
14 ms |
7424 KB |
Output is correct |
90 |
Correct |
14 ms |
7040 KB |
Output is correct |
91 |
Correct |
28 ms |
8320 KB |
Output is correct |
92 |
Correct |
27 ms |
8184 KB |
Output is correct |
93 |
Correct |
4 ms |
5120 KB |
Output is correct |
94 |
Correct |
147 ms |
35184 KB |
Output is correct |
95 |
Correct |
137 ms |
34932 KB |
Output is correct |
96 |
Correct |
139 ms |
34296 KB |
Output is correct |
97 |
Correct |
144 ms |
34912 KB |
Output is correct |
98 |
Correct |
169 ms |
35176 KB |
Output is correct |
99 |
Correct |
296 ms |
25464 KB |
Output is correct |
100 |
Correct |
274 ms |
25208 KB |
Output is correct |
101 |
Correct |
294 ms |
25208 KB |
Output is correct |
102 |
Correct |
259 ms |
25720 KB |
Output is correct |
103 |
Correct |
170 ms |
29176 KB |
Output is correct |
104 |
Correct |
182 ms |
29436 KB |
Output is correct |
105 |
Correct |
172 ms |
29176 KB |
Output is correct |
106 |
Correct |
197 ms |
30124 KB |
Output is correct |
107 |
Correct |
192 ms |
29932 KB |
Output is correct |
108 |
Correct |
149 ms |
34800 KB |
Output is correct |
109 |
Correct |
165 ms |
34808 KB |
Output is correct |
110 |
Correct |
213 ms |
25672 KB |
Output is correct |
111 |
Correct |
219 ms |
25720 KB |
Output is correct |
112 |
Correct |
375 ms |
24824 KB |
Output is correct |
113 |
Correct |
273 ms |
24824 KB |
Output is correct |