0
点赞
收藏
分享

微信扫一扫

洛谷P4913 二叉树深度

胡桑_b06e 2022-03-13 阅读 63

题目描述

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

最多有 10^6106 个结点。

输入格式

第一行一个整数 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

就是一个典型的二叉树加深搜,但是想到python标准的二叉树写起来非常麻烦,看了C++的题解,用列表来存每一个‘树类’,每个树类的左右儿子都只是放那个树类在列表里的序号,方便深搜时按序号找

class Tree:    # 创建树类
    def __init__(self):
        self.lchild = None
        self.rchild = None

n = int(input())
lst = [Tree() for i in range(n)]      # 列表里的每一个元素都是一个树
for i in range(n):              # 记录每一个树的左右儿子在列表里的位置
    a,b = map(int,input().split())      # 如果输进来的是0,存进去的是-1,识别时要注意
    lst[i].lchild = a-1
    lst[i].rchild = b-1
num1 = 1
def DFS(point,num):             # 参数是列表里的位置,当前深度
    global num1
    num1 = max(num1,num)            # 记录到目前为止最大深度
    if lst[point].lchild == -1:
        return
    else:
        DFS(lst[point].lchild,num+1)
        DFS(lst[point].rchild,num+1)
DFS(0,1)
print(num1)
举报

相关推荐

0 条评论