【洛谷P4913 深基16.例3】二叉树深度
题目描述
有一个
n
(
n
≤
1
0
6
)
n(n \le 10^6)
n(n≤106) 个结点的二叉树。给出每个结点的两个子结点编号(均不超过
n
n
n),建立一棵二叉树(根节点的编号为
1
1
1),如果是叶子结点,则输入 0 0
。
建好这棵二叉树之后,请求出它的深度。二叉树的深度是指从根节点到叶子结点时,最多经过了几层。
输入格式
第一行一个整数 n n n,表示结点数。
之后 n n n 行,第 i i i 行两个整数 l l l、 r r r,分别表示结点 i i i 的左右子结点编号。若 l = 0 l=0 l=0 则表示无左子结点, r = 0 r=0 r=0 同理。
输出格式
一个整数,表示最大结点深度。
样例 #1
样例输入 #1
7
2 7
3 6
4 5
0 0
0 0
0 0
0 0
样例输出 #1
4
正确代码
#include <bits/stdc++.h>
#include <algorithm>
using namespace std;
int N;
const int MAXN = 1e6+5;
struct node {
int l,r;
};
void dfs(int num, int deep);
node tree[MAXN];
int ans = 0;
int main() {
cin >> N;
for (int i=1; i<=N; i++) {
cin >> tree[i].l >> tree[i].r;
}
dfs(1,1);
cout << ans << endl;
return 0;
}
void dfs(int num, int deep) {
if (num == 0) return ;
ans = max(ans,deep);
dfs(tree[num].l,deep+1);
dfs(tree[num].r,deep+1);
}