版权声明:本文为博主原创文章,未经博主允许不得转载。转载请注明作者和出处:https://blog.csdn.net/weixin_41481113 https://blog.csdn.net/weixin_41481113/article/details/83588736
题目描述
现在有一棵合法的二叉树,树的节点都是用数字表示,现在给定这棵树上所有的父子关系,求这棵树的高度
输入描述:
输入的第一行表示节点的个数n(1 ≤ n ≤ 1000,节点的编号为0到n-1)组成,
下面是n-1行,每行有两个整数,第一个数表示父节点的编号,第二个数表示子节点的编号
输出描述:
输出树的高度,为一个整数
示例1
输入
5
0 1
0 2
1 3
1 4
输出
3
代码实现
import sys
lines = sys.stdin.readlines()
N=int(lines[0])
node=[]
height=[0]*N
height[0]=1
child_num=[0]*N
for i in range(1,N):
node.append(tuple(lines[i].split()))
for i in range(N-1):
if child_num[int(node[i][0])]<2:
height[i+1]=height[int(node[i][0])]+1
child_num[int(node[i][0])]+=1
print(max(height))