백준2250 트리의 높이와 너비

작성일 | In

문제 링크

  • http://icpc.me/2250

문제 출처

  • 2003 전국 본선 고등부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
#include <bits/stdc++.h>
using namespace std;

int mn[10101];
int mx[10101];
int par[10101];
int cnt;
vector<int> g[10101];

void dfs(int now, int dep){
	if(g[now][0] != -1) dfs(g[now][0], dep+1);
	cnt++;
	mn[dep] = min(mn[dep], cnt);
	mx[dep] = max(mx[dep], cnt);
	if(g[now][1] != -1) dfs(g[now][1], dep+1);
}

int main(){
	ios_base::sync_with_stdio(0); cin.tie(0);
	memset(mn, 0x3f, sizeof mn);
	memset(mx, 0, sizeof mx);
	memset(par, -1, sizeof par);

	int n; cin >> n;
	for(int i=1; i<=n; i++){
		int a, b, c; cin >> a >> b >> c;
		g[a].push_back(b);
		g[a].push_back(c);
		if(b != -1) par[b] = a;
		if(c != -1) par[c] = a;
	}
	for(int i=1; i<=n; i++){
		if(par[i] == -1) dfs(i, 1);
	}
	int idx = 1;
	for(int i=2; i<10101; i++){
		if(mn[i] <= mx[i]){
			if(mx[i] - mn[i] + 1 > mx[idx] - mn[idx] + 1) idx = i;
		}
	}
	cout << idx << " " << (mx[idx] - mn[idx] + 1);
}