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

题目描述

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

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

输入格式

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

之后 nn 行，第 ii 行两个整数 ll、rr，分别表示结点 ii 的左右子结点编号。若 l=0l=0 则表示无左子结点，r=0r=0 同理。

输出格式

一个整数，表示最大结点深度。

输入输出样例

输入 #1复制

7
2 7
3 6
4 5
0 0
0 0
0 0
0 0

输出 #1复制

4

#include<bits/stdc++.h>
using namespace std;
const int N=1e6+2;
int a[N],b[N];		//a:左子树，b：右子树；
int n;
int shen(int gen){
	if(!gen){
		return 0;
	}
	int h1=shen(a[gen]);
	int h2=shen(b[gen]);
	return 1+max(h1,h2);
} 
int main(){
	ios::sync_with_stdio(false);
	cin>>n;
	int l,r;
	for(int i=1;i<+n;i++){
		cin>>l>>r;
		a[i]=l;
		b[i]=r;
	}
	int res=shen(1);
	cout<<res<<endl;
	return 0;
}