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

题目描述

给出每个节点的两个儿子节点，建立一棵二叉树（根节点为 1），如果是叶子节点，则输入0 0。建好树后希望知道这棵二叉树的深度。二叉树的深度是指从根节点到叶子结点时，最多经过了几层。

最多有 10^6 个结点。

输入格式

无

输出格式

无

输入输出样例

输入 #1

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

输出 #1

4

建树也有一个技巧，我们可以使用结构体数组分别存储左右节点，因为这样刚好符合题意，第一个节点收第一行的左右子树，即下标为1的结构体数组，以此类推。

关键问题是，建好二叉树后，该如何计算最大的层数呢？我们可以使用dfs。从总体来讲，我们寻找最大层数就是从根节点开始，然后找左子树，然后找右子树，而左子树中可能还会有左子树右子树，根节点的右子树也是如此。完全可以使用dfs，使用递归的方式计算最大层数。具体实现代码如下：

#include <stdio.h>

#include <stdlib.h>

struct node{
    int l,r;
}t[1000010];
int ans=-1;
void dfs(int x,int cnt)
{
    if(t[x].l==0&&t[x].r==0)//已知若左右子节点为0则代表是根节点
        return ;//此时已经是最底层的”下一层“，即空，返回
    ans=ans>cnt?ans:cnt;//每到一层就实时更新最大层数
    dfs(t[x].l,cnt+1);//走左子树，寻找最多层
    dfs(t[x].r,cnt+1);//走右子树，寻找最多层
}
int main()
{
    int n;
    scanf("%d",&n);
    for(int i=1;i<=n;i++)
    scanf("%d %d",&t[i].l,&t[i].r);
    dfs(1,2);//1即代表第一个节点，2即代表传进去后判断的是第二层的数
    printf("%d",ans);
}