练习题二:树的高度
题目描述
现在有一棵合法的二叉树,树的节点都是用数字表示,现在给定这棵树上所有的父子关系,求这棵树的高度
输入描述:
输入的第一行表示节点的个数n(1 ≤ n ≤ 1000,节点的编号为0到n-1)组成,
下面是n-1行,每行有两个整数,第一个数表示父节点的编号,第二个数表示子节点的编号。
输出描述:
输出树的高度,为一个整数
Python解法
# -*- coding:utf-8 -*-
import sys
def tree_height(root):
if len(tree[root]) == 0:
return 1
node_list = []
for i in tree[root]:
node_list.append(tree_height(i))
return max(node_list) + 1
if __name__ == "__main__":
n = sys.stdin.readline().strip()
tree = []
for i in range(int(n)):
tree.append([])
for i in range(int(n)-1):
father, child = map(int, sys.stdin.readline().strip().split())
if len(tree[father]) <= 1 :
tree[father].append(child)
print(tree_height(0))
上一篇: 教你用3DSMAX简单制作弯曲的马路贴图
下一篇: python全栈工程师