반응형
https://www.acmicpc.net/problem/15681
15681번: 트리와 쿼리
트리의 정점의 수 N과 루트의 번호 R, 쿼리의 수 Q가 주어진다. (2 ≤ N ≤ 105, 1 ≤ R ≤ N, 1 ≤ Q ≤ 105) 이어 N-1줄에 걸쳐, U V의 형태로 트리에 속한 간선의 정보가 주어진다. (1 ≤ U, V ≤ N, U ≠ V)
www.acmicpc.net
DFS로 간단하게 해결 가능
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
|
#include <iostream>
#include <vector>
#define rep(i,n) for(int i=1;i<=n;i++)
using namespace std;
vector<vector<int> > v;
int n, r, q, a, b,x, child[100001], visited[100001];
void input();
void print();
int func(int r) {
for (int next : v[r]) {
if (!visited[next]) {
visited[next] = 1;
child[r]+=func(next);
}
}
return child[r] + 1;
}
int main() {
ios_base::sync_with_stdio(false);
cin.tie(NULL);
input();
func(r);
print();
}
void input() {
cin >> n >> r >> q;
v.resize(n + 1);
rep(i, n - 1) {
cin >> a >> b;
v[a].push_back(b);
v[b].push_back(a);
}
visited[r] = 1;
}
void print() {
rep(i, q) {
cin >> x;
cout << child[x] + 1 << '\n';
}
}
|
cs |
반응형
'백준 > 트리DP' 카테고리의 다른 글
| 백준 2213 (0) | 2021.08.02 |
|---|---|
| 백준 2533 [복습필수] (0) | 2021.08.01 |
| 백준 1005 (0) | 2021.07.24 |