CCF PTA 2023年5月C++天空之城的树

全栈工程师Linda2024-05-05 7:08

【问题描述】

拉姆达人在修建天空之城时，主要是依赖巨大的飞行石去维持悬空状态，依赖强壮的大树去作为建筑物的框架，假设大树是一棵有 n(n≤10 3)个结点的二叉树。给出每个结点的两个子结点编号（均不超过 n），建立一棵二叉树（根节点的编号为 1），如果是叶子结点，则输入 0 0。建好这棵二叉树之后，请帮拉姆达设计师求出它的深度。二叉树的深度是指从根节点到叶子结点时，最多经过了几层。

【输入描述】

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

【输出描述】

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

【输入样例】

2 7

3 6

4 5

0 0

【题解】

本题关键点：用递归的方式求二叉树的深度。代码如下。

cpp 复制代码

#include <iostream>
using namespace std;
struct tree{
	int left;
	int right;
};
tree all[1010];
int depth=0;
//root当前的根，d当前的深度 
void dep(int root,int d){
	if(root==0)
		return;
	depth=max(depth,d);
	dep(all[root].left,d+1);
	dep(all[root].right,d+1);
}
int main(){
	int n;	
	cin>>n;
	for(int i=1;i<=n;i++){
		cin>>all[i].left>>all[i].right;		
	}	
	dep(1,1);
	cout<<depth<<endl;
	return 0;
}