Untitled

mail@pastecode.io avatar
unknown
plain_text
6 months ago
885 B
1
Indexable
Never
# class Tree:
#     def __init__(self, number, input_list):
#         self.number = int(number)
#         self.input_list = list(map(int, input.split()))
#         temp_dict = {}
#         for i in range(len(self.number)):
#             temp_dict.update({i:self.input_list[i]})
#         self.tree_dict = temp_dict

def depth(tree, i):
    # Если родителем вершины является она сама, то глубина равна 1
    if tree.tree_dict[i] == -1:
        return 1
    else:
        # иначе глубина равна глубине родителя + 1
        return 1 + depth(tree, tree.tree_dict[i])
    

def solve(tree):
    if tree.number > 0:
        heights = [0]*tree.number
        for i in range(tree.number):
            heights[i] = depth(tree, i)
        return max(heights)-1
    else:
        return 0
Leave a Comment