P4913 【深基16.例3】二叉树深度
题目描述
有一个 n(n≤106) 个结点的二叉树。给出每个结点的两个子结点编号(均不超过 n),建立一棵二叉树(根节点的编号为 11),如果是叶子结点,则输入 0 0
。
建好这棵二叉树之后,请求出它的深度。二叉树的深度是指从根节点到叶子结点时,最多经过了几层。
输入格式
第一行一个整数 n,表示结点数。
之后 n 行,第 i 行两个整数 l、r,分别表示结点 i 的左右子结点编号。若 l=0 则表示无左子结点,r=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;
int max1=0;//初始化max1为0,好做最大值的比较
struct point{//定义结构体,将左右子树存于结构体中
int l;
int r;
}ch[10000000];//定义结构体数组
void dfs(int n,int step)//深搜函数,n为节点编号,step为层数
{
if(step>=max1){//如果step大于max1,则将step的值赋值给max1
max1=step;
}
if(ch[n].l!=NULL){//当左子树不为空时,选择左边进行深搜
dfs(ch[n].l,step+1);//搜索进入下一层,且step+1
}
if(ch[n].r!=NULL){//当右子树不为空时,选择右边进行深搜
dfs(ch[n].r,step+1);//搜索进入下一层,且step+1
}
return;
}
int main()
{
int n;
scanf("%d",&n);
for(int i=1;i<=n;i++){
scanf("%d %d",&ch[i].l,&ch[i].r);//输入i节点的左右子树
}
int step;//定义层数,并且初始化为1
dfs(1,1);//
printf("%d",max1);
return 0;
}
这题是一个与二叉树有一丢丢关系的dfs题。该题主要就是通过dfs的思想,选一边搜索到底,在进行回溯,又从另一边搜索到底,最后比较各step值,从而得出最大值。