문제
https://www.acmicpc.net/problem/19542
문제해설
2020 ucpc 본선 A번 문제였다.
전단지를 배달하는 현민이가 있다.
그래프가 주어지고 시작할 노드의 번호와 D값이 주어진다.
D값은 정확히 정의가 안되어있는데
직접 이동하지 않고 전단지를 날렸을 때 도달할 수 있는 거리라고 생각하면 될 것 같다.
문제풀이
잘하는 사람들은 엄청 금방금방 풀던데 나는 엄청 오래 고민했던 문제이다.
dfs를 이용해야 된다는 것은 자명해보이는데
자식들이 나누어지는 부분에서 처리하는게 까다로웠다.
이번에 이 문제를 많이 고민하면서 재귀와 좀 더 친해진 느낌이다.
나는 dfs를 2번 해서 풀었다.
우선 첫번째 dfs는 리프노드까지 내려간 후
다시 올라오면서 0부터 시작해서 ++해주면서 노드에 번호를 붙였다.
이런식으로 저장이 된다.
그리고 2번이나 1번 노드에서는 두 값중 최대값을 저장했다.
두번째 dfs를 돌 때는 해당 배열의 값이
D보다 크거나 같을 때만 해당 노드를 방문하고
(방문한 총 노드의 개수 * 2) - 2가 정답이 된다.
그리고 그래프를 단방향으로만 만들면
시작노드가 1이 아닐 때 제대로 안나오기 때문에
꼭 양방향 그래프로 만들어줘야 한다.
잘 짜면 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
41
42
43
44
45
46
47
48
49
50
#include<iostream>
#include<vector>
#include<algorithm>
using namespace std;
int n, s, d, res;
vector<vector<int>>arr;
int cnt[100001], visited[100001];
void go(int now) {
for (int i = 0; i < arr[now].size(); i++) {
int next = arr[now][i];
if (cnt[next] >= d && !visited[next]) {
res++;
visited[next] = 1;
go(next);
}
}
}
int dfs(int now) {
int dep = 0;
for (int i = 0; i < arr[now].size(); i++) {
int next = arr[now][i];
if (visited[next]) continue;
visited[next] = 1;
dep = dfs(next);
cnt[now] = max(cnt[now], dep);
}
return max(cnt[now], dep) + 1;
}
int main() {
ios::sync_with_stdio(false); cin.tie(NULL); cout.tie(NULL);
cin >> n >> s >> d;
arr.resize(n + 1);
for (int i = 0; i < n - 1; i++) {
int a, b; cin >> a >> b;
arr[a].push_back(b);
arr[b].push_back(a);
}
visited[s] = 1;
dfs(s);
fill(visited, visited + 100001, 0);
visited[s] = 1;
go(s);
cout << (res + 1) * 2 - 2;
return 0;
}