【深基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
建树
class tree{
public:
int left,right;
}trees[maxn];
插入
int n;
cin>>n;
for(int i = 1; i<=n;i++)
{
cin>>trees[i].left>>trees[i].right;
}
关键 dfs
如果遇到叶子(id=0)则返回,否则继续搜索左和右。深度+1
void dfs(int id,int depth)
{
if(id==0)
{
return ;
}
cnt=max(cnt,depth);
dfs(trees[id].left,depth+1);
dfs(trees[id].right,depth+1);
}
AC代码
#include<bits/stdc++.h>
using namespace std;
const int maxn = 1e6+5;
int cnt=0;
class tree{
public:
int left,right;
}trees[maxn];
void dfs(int id,int depth)
{
if(id==0)
{
return ;
}
cnt=max(cnt,depth);
dfs(trees[id].left,depth+1);
dfs(trees[id].right,depth+1);
}
int main()
{
int n;
cin>>n;
for(int i = 1; i<=n;i++)
{
cin>>trees[i].left>>trees[i].right;
}
dfs(1,1);
cout<<cnt;
return 0;
}