///~~~LOTA~~~///
#include <bits/stdc++.h>
using namespace std;
#define ll long long
#define append push_back
#define add insert
#define nl "\n"
#define ff first
#define ss second
#define pii pair<int,int>
#define all(x) (x).begin(),(x).end()
#define L0TA ios_base::sync_with_stdio(false);cin.tie(NULL)
#define N 100001
int k;
int s[N];
vector<int> ans;
vector<pii> a[N];
vector<int> c[N];
map<int,int> *x[N];
void dfs(int v,int u){
int z=0;
for(auto& i:a[v]){
if(i.ss==u) continue;
dfs(i.ff,i.ss);
if((*x[i.ff]).size()>(*x[z]).size())
z=i.ff;
}
if(z>0) x[v]=x[z];
else x[v]=new map<int,int>;
for(auto &i:a[v]){
if(i.ss==u || i.ff==z)
continue;
for(auto& j:*x[i.ff]){
(*x[v])[j.ff]+=j.ss;
if(s[j.ff]==(*x[v])[j.ff])
(*x[v]).erase(j.ff);
}
(*x[i.ff]).clear();
}
for(auto& i:c[v]){
(*x[v])[i]++;
if(s[i]==(*x[v])[i])
(*x[v]).erase(i);
}
if((*x[v]).size()>=k)
ans.append(u);
}
void solve(){
int n,m,p,q;
cin>>n>>m>>k;
for(int i=1;i<n;i++){
cin>>p>>q;
a[p].append({q,i});
a[q].append({p,i});
}
for(int i=0;i<m;i++){
cin>>s[i];
for(int j=0;j<s[i];j++){
cin>>q;
c[q].append(i);
}
}
x[0]=new map<int,int>;
dfs(1,0);
sort(all(ans));
cout<<ans.size()<<nl;
for(auto& i:ans) cout<<i<<' ';
}
int main(){
L0TA;
solve();
return 0;
}
Compilation message
railway.cpp: In function 'void dfs(int, int)':
railway.cpp:45:22: warning: comparison of integer expressions of different signedness: 'std::map<int, int>::size_type' {aka 'long unsigned int'} and 'int' [-Wsign-compare]
45 | if((*x[v]).size()>=k)
| ~~~~~~~~~~~~~~^~~
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
1 ms |
5724 KB |
Output is correct |
2 |
Correct |
4 ms |
6748 KB |
Output is correct |
3 |
Correct |
4 ms |
6808 KB |
Output is correct |
4 |
Correct |
2 ms |
5776 KB |
Output is correct |
5 |
Correct |
1 ms |
5948 KB |
Output is correct |
6 |
Correct |
5 ms |
7260 KB |
Output is correct |
7 |
Correct |
6 ms |
6748 KB |
Output is correct |
8 |
Correct |
4 ms |
6976 KB |
Output is correct |
9 |
Correct |
5 ms |
7004 KB |
Output is correct |
10 |
Correct |
2 ms |
5724 KB |
Output is correct |
11 |
Correct |
2 ms |
5724 KB |
Output is correct |
12 |
Correct |
2 ms |
5724 KB |
Output is correct |
13 |
Correct |
2 ms |
5720 KB |
Output is correct |
14 |
Correct |
1 ms |
5724 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
1 ms |
5724 KB |
Output is correct |
2 |
Correct |
4 ms |
6748 KB |
Output is correct |
3 |
Correct |
4 ms |
6808 KB |
Output is correct |
4 |
Correct |
2 ms |
5776 KB |
Output is correct |
5 |
Correct |
1 ms |
5948 KB |
Output is correct |
6 |
Correct |
5 ms |
7260 KB |
Output is correct |
7 |
Correct |
6 ms |
6748 KB |
Output is correct |
8 |
Correct |
4 ms |
6976 KB |
Output is correct |
9 |
Correct |
5 ms |
7004 KB |
Output is correct |
10 |
Correct |
2 ms |
5724 KB |
Output is correct |
11 |
Correct |
2 ms |
5724 KB |
Output is correct |
12 |
Correct |
2 ms |
5724 KB |
Output is correct |
13 |
Correct |
2 ms |
5720 KB |
Output is correct |
14 |
Correct |
1 ms |
5724 KB |
Output is correct |
15 |
Correct |
24 ms |
7652 KB |
Output is correct |
16 |
Correct |
23 ms |
7760 KB |
Output is correct |
17 |
Correct |
23 ms |
7764 KB |
Output is correct |
18 |
Correct |
6 ms |
7260 KB |
Output is correct |
19 |
Correct |
5 ms |
6744 KB |
Output is correct |
20 |
Correct |
31 ms |
12888 KB |
Output is correct |
21 |
Correct |
29 ms |
9664 KB |
Output is correct |
22 |
Correct |
1 ms |
5720 KB |
Output is correct |
23 |
Correct |
5 ms |
6804 KB |
Output is correct |
24 |
Correct |
4 ms |
6748 KB |
Output is correct |
25 |
Correct |
1 ms |
5768 KB |
Output is correct |
26 |
Correct |
1 ms |
5724 KB |
Output is correct |
27 |
Correct |
5 ms |
7348 KB |
Output is correct |
28 |
Correct |
4 ms |
6748 KB |
Output is correct |
29 |
Correct |
4 ms |
7004 KB |
Output is correct |
30 |
Correct |
5 ms |
7004 KB |
Output is correct |
31 |
Correct |
2 ms |
5724 KB |
Output is correct |
32 |
Correct |
2 ms |
5768 KB |
Output is correct |
33 |
Correct |
2 ms |
5724 KB |
Output is correct |
34 |
Correct |
1 ms |
5724 KB |
Output is correct |
35 |
Correct |
2 ms |
5724 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
81 ms |
25548 KB |
Output is correct |
2 |
Correct |
2 ms |
5720 KB |
Output is correct |
3 |
Correct |
83 ms |
27104 KB |
Output is correct |
4 |
Correct |
87 ms |
26704 KB |
Output is correct |
5 |
Correct |
63 ms |
26436 KB |
Output is correct |
6 |
Correct |
63 ms |
26448 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
68 ms |
21408 KB |
Output is correct |
2 |
Correct |
96 ms |
19440 KB |
Output is correct |
3 |
Correct |
99 ms |
18232 KB |
Output is correct |
4 |
Correct |
92 ms |
22948 KB |
Output is correct |
5 |
Correct |
108 ms |
21140 KB |
Output is correct |
6 |
Correct |
81 ms |
23768 KB |
Output is correct |
7 |
Correct |
73 ms |
24920 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
68 ms |
21408 KB |
Output is correct |
2 |
Correct |
96 ms |
19440 KB |
Output is correct |
3 |
Correct |
99 ms |
18232 KB |
Output is correct |
4 |
Correct |
92 ms |
22948 KB |
Output is correct |
5 |
Correct |
108 ms |
21140 KB |
Output is correct |
6 |
Correct |
81 ms |
23768 KB |
Output is correct |
7 |
Correct |
73 ms |
24920 KB |
Output is correct |
8 |
Correct |
64 ms |
22152 KB |
Output is correct |
9 |
Correct |
77 ms |
21844 KB |
Output is correct |
10 |
Correct |
59 ms |
26104 KB |
Output is correct |
11 |
Correct |
59 ms |
26208 KB |
Output is correct |
12 |
Correct |
58 ms |
19916 KB |
Output is correct |
13 |
Correct |
59 ms |
20080 KB |
Output is correct |
14 |
Correct |
77 ms |
18928 KB |
Output is correct |
15 |
Correct |
73 ms |
18752 KB |
Output is correct |
16 |
Correct |
105 ms |
22660 KB |
Output is correct |
17 |
Correct |
117 ms |
19816 KB |
Output is correct |
18 |
Correct |
92 ms |
23068 KB |
Output is correct |
19 |
Correct |
88 ms |
18824 KB |
Output is correct |
20 |
Correct |
76 ms |
23636 KB |
Output is correct |
21 |
Correct |
80 ms |
23740 KB |
Output is correct |
22 |
Correct |
71 ms |
24372 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
1 ms |
5724 KB |
Output is correct |
2 |
Correct |
4 ms |
6748 KB |
Output is correct |
3 |
Correct |
4 ms |
6808 KB |
Output is correct |
4 |
Correct |
2 ms |
5776 KB |
Output is correct |
5 |
Correct |
1 ms |
5948 KB |
Output is correct |
6 |
Correct |
5 ms |
7260 KB |
Output is correct |
7 |
Correct |
6 ms |
6748 KB |
Output is correct |
8 |
Correct |
4 ms |
6976 KB |
Output is correct |
9 |
Correct |
5 ms |
7004 KB |
Output is correct |
10 |
Correct |
2 ms |
5724 KB |
Output is correct |
11 |
Correct |
2 ms |
5724 KB |
Output is correct |
12 |
Correct |
2 ms |
5724 KB |
Output is correct |
13 |
Correct |
2 ms |
5720 KB |
Output is correct |
14 |
Correct |
1 ms |
5724 KB |
Output is correct |
15 |
Correct |
24 ms |
7652 KB |
Output is correct |
16 |
Correct |
23 ms |
7760 KB |
Output is correct |
17 |
Correct |
23 ms |
7764 KB |
Output is correct |
18 |
Correct |
6 ms |
7260 KB |
Output is correct |
19 |
Correct |
5 ms |
6744 KB |
Output is correct |
20 |
Correct |
31 ms |
12888 KB |
Output is correct |
21 |
Correct |
29 ms |
9664 KB |
Output is correct |
22 |
Correct |
1 ms |
5720 KB |
Output is correct |
23 |
Correct |
5 ms |
6804 KB |
Output is correct |
24 |
Correct |
4 ms |
6748 KB |
Output is correct |
25 |
Correct |
1 ms |
5768 KB |
Output is correct |
26 |
Correct |
1 ms |
5724 KB |
Output is correct |
27 |
Correct |
5 ms |
7348 KB |
Output is correct |
28 |
Correct |
4 ms |
6748 KB |
Output is correct |
29 |
Correct |
4 ms |
7004 KB |
Output is correct |
30 |
Correct |
5 ms |
7004 KB |
Output is correct |
31 |
Correct |
2 ms |
5724 KB |
Output is correct |
32 |
Correct |
2 ms |
5768 KB |
Output is correct |
33 |
Correct |
2 ms |
5724 KB |
Output is correct |
34 |
Correct |
1 ms |
5724 KB |
Output is correct |
35 |
Correct |
2 ms |
5724 KB |
Output is correct |
36 |
Correct |
81 ms |
25548 KB |
Output is correct |
37 |
Correct |
2 ms |
5720 KB |
Output is correct |
38 |
Correct |
83 ms |
27104 KB |
Output is correct |
39 |
Correct |
87 ms |
26704 KB |
Output is correct |
40 |
Correct |
63 ms |
26436 KB |
Output is correct |
41 |
Correct |
63 ms |
26448 KB |
Output is correct |
42 |
Correct |
68 ms |
21408 KB |
Output is correct |
43 |
Correct |
96 ms |
19440 KB |
Output is correct |
44 |
Correct |
99 ms |
18232 KB |
Output is correct |
45 |
Correct |
92 ms |
22948 KB |
Output is correct |
46 |
Correct |
108 ms |
21140 KB |
Output is correct |
47 |
Correct |
81 ms |
23768 KB |
Output is correct |
48 |
Correct |
73 ms |
24920 KB |
Output is correct |
49 |
Correct |
64 ms |
22152 KB |
Output is correct |
50 |
Correct |
77 ms |
21844 KB |
Output is correct |
51 |
Correct |
59 ms |
26104 KB |
Output is correct |
52 |
Correct |
59 ms |
26208 KB |
Output is correct |
53 |
Correct |
58 ms |
19916 KB |
Output is correct |
54 |
Correct |
59 ms |
20080 KB |
Output is correct |
55 |
Correct |
77 ms |
18928 KB |
Output is correct |
56 |
Correct |
73 ms |
18752 KB |
Output is correct |
57 |
Correct |
105 ms |
22660 KB |
Output is correct |
58 |
Correct |
117 ms |
19816 KB |
Output is correct |
59 |
Correct |
92 ms |
23068 KB |
Output is correct |
60 |
Correct |
88 ms |
18824 KB |
Output is correct |
61 |
Correct |
76 ms |
23636 KB |
Output is correct |
62 |
Correct |
80 ms |
23740 KB |
Output is correct |
63 |
Correct |
71 ms |
24372 KB |
Output is correct |
64 |
Correct |
70 ms |
22608 KB |
Output is correct |
65 |
Correct |
135 ms |
19396 KB |
Output is correct |
66 |
Correct |
97 ms |
17744 KB |
Output is correct |
67 |
Correct |
87 ms |
17904 KB |
Output is correct |
68 |
Correct |
46 ms |
18856 KB |
Output is correct |
69 |
Correct |
43 ms |
19028 KB |
Output is correct |
70 |
Correct |
87 ms |
23248 KB |
Output is correct |
71 |
Correct |
63 ms |
22548 KB |
Output is correct |
72 |
Correct |
2 ms |
5724 KB |
Output is correct |
73 |
Correct |
5 ms |
6724 KB |
Output is correct |
74 |
Correct |
5 ms |
6748 KB |
Output is correct |
75 |
Correct |
2 ms |
5724 KB |
Output is correct |
76 |
Correct |
1 ms |
5724 KB |
Output is correct |
77 |
Correct |
5 ms |
7292 KB |
Output is correct |
78 |
Correct |
5 ms |
7004 KB |
Output is correct |
79 |
Correct |
4 ms |
6920 KB |
Output is correct |
80 |
Correct |
5 ms |
7256 KB |
Output is correct |
81 |
Correct |
2 ms |
5976 KB |
Output is correct |
82 |
Correct |
2 ms |
5764 KB |
Output is correct |
83 |
Correct |
2 ms |
5720 KB |
Output is correct |
84 |
Correct |
1 ms |
5724 KB |
Output is correct |
85 |
Correct |
2 ms |
5720 KB |
Output is correct |
86 |
Correct |
23 ms |
7764 KB |
Output is correct |
87 |
Correct |
23 ms |
7648 KB |
Output is correct |
88 |
Correct |
24 ms |
7680 KB |
Output is correct |
89 |
Correct |
7 ms |
7260 KB |
Output is correct |
90 |
Correct |
6 ms |
7000 KB |
Output is correct |
91 |
Correct |
31 ms |
12888 KB |
Output is correct |
92 |
Correct |
30 ms |
9816 KB |
Output is correct |
93 |
Correct |
2 ms |
5724 KB |
Output is correct |
94 |
Correct |
81 ms |
27380 KB |
Output is correct |
95 |
Correct |
77 ms |
27004 KB |
Output is correct |
96 |
Correct |
78 ms |
26700 KB |
Output is correct |
97 |
Correct |
65 ms |
26356 KB |
Output is correct |
98 |
Correct |
68 ms |
26656 KB |
Output is correct |
99 |
Correct |
104 ms |
22864 KB |
Output is correct |
100 |
Correct |
107 ms |
22540 KB |
Output is correct |
101 |
Correct |
112 ms |
22864 KB |
Output is correct |
102 |
Correct |
78 ms |
19024 KB |
Output is correct |
103 |
Correct |
70 ms |
24036 KB |
Output is correct |
104 |
Correct |
79 ms |
24332 KB |
Output is correct |
105 |
Correct |
68 ms |
23028 KB |
Output is correct |
106 |
Correct |
65 ms |
22872 KB |
Output is correct |
107 |
Correct |
74 ms |
22828 KB |
Output is correct |
108 |
Correct |
59 ms |
26112 KB |
Output is correct |
109 |
Correct |
59 ms |
26264 KB |
Output is correct |
110 |
Correct |
48 ms |
20048 KB |
Output is correct |
111 |
Correct |
49 ms |
20108 KB |
Output is correct |
112 |
Correct |
87 ms |
19028 KB |
Output is correct |
113 |
Correct |
89 ms |
18624 KB |
Output is correct |