左孩子右兄弟 蓝桥杯1451 python

题目描述

对于一棵多叉树，我们可以通过 “左孩子右兄弟” 表示法，将其转化成一棵二叉树。

如果我们认为每个结点的子结点是无序的，那么得到的二叉树可能不唯一。

换句话说，每个结点可以选任意子结点作为左孩子，并按任意顺序连接右兄弟。

给定一棵包含 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 = 0
  for 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])