二叉树深度

【深基16.例3】二叉树深度

题目描述

有一个 $n(n\le 10^6)$ 个结点的二叉树。给出每个结点的两个子结点编号（均不超过 $n$），建立一棵二叉树（根节点的编号为 $1$），如果是叶子结点，则输入 0 0。

建好这棵二叉树之后，请求出它的深度。二叉树的深度是指从根节点到叶子结点时，最多经过了几层。

输入格式

第一行一个整数 $n$，表示结点数。

之后 $n$ 行，第 $i$ 行两个整数 $l$、$r$，分别表示结点 $i$ 的左右子结点编号。若 $l=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;
}