DFS遍历 | 左孩子右兄弟二叉树

73分，时间超限

n=int(input())

tree=[[] for _ in range(n+1)]

for i in range(2,n+1):

    parent=int(input())

    tree[parent].append(i)

def dfs(node):

    if len(tree[node])==0:

        return 0

    max_height=0

    for i in tree[node]:#对各个孩子结点的高进行比较，选出最大的

        max_height=max(max_height,dfs(i))

    return max_height+len(tree[node])#孩子结点的高+自己+兄弟结点的数量

height=dfs(1)

print(height)