题目描述
对于一棵多叉树,我们可以通过 “左孩子右兄弟” 表示法,将其转化成一棵二叉树。
如果我们认为每个结点的子结点是无序的,那么得到的二叉树可能不唯一。
换句话说,每个结点可以选任意子结点作为左孩子,并按任意顺序连接右兄弟。
给定一棵包含 N 个结点的多叉树,结点从 1 至 N 编号,其中 1 号结点是根,每个结点的父结点的编号比自己的编号小。
请你计算其通过 “左孩子右兄弟” 表示法转化成的二叉树,高度最高是多少。
注:只有根结点这一个结点的树高度为 0。
输入描述
输入的第一行包含一个整数 N。 以下 N−1 行,每行包含一个整数,依次表示 2 至 N 号结点的父结点编号。
输出描述
输出一个整数表示答案。
输入输出样例
示例 1
输入
5
1
1
1
2
输出
4
思路:创建一个邻接表,把每一行输入的数字挨个存进去,最大深度就是孩子的数量,加上以这个孩子节点为父节点的孩子数量,以此类推
# 树形DP
g = [[] for j in range(1, 100010)] # 邻接表
dp = [0] * 100010
def dfs(u):dp[u] = len(g[u]) # 全部变为左孩子,那么长度就是孩子的数量maxv = 0for v in g[u]:#以当前孩子节点为父节点,寻找下一个孩子节点dfs(v) # DFS孩子maxv = max(dp[v], maxv) # 取最大深度dp[u] += maxv # 每次加上最大深度
n = int(input())
for i in range(2, n + 1):v = int(input())g[v].append(i)
dfs(1)
print(dp[1])