python Высотадеревазадонногочислами

Posted 2021-05-09

import sys
from collections import defaultdict

input = sys.stdin
# input = open('2-tree.txt', 'r')
n, text = input.readline(), [int(i) for i in input.readline().split()]

tree = defaultdict(list)

for i, l in enumerate(text):
    tree[l].append(i)

root = tree[-1][0]


def height(root):
    h = 1
    for v in tree[root]:
        h = max(h, height(v) + 1)
    return h


print(height(root))