Laskettelukeskus

class Tree:
    def __init__(self, value=None, children=None):
        self.value = value
        self.children = []
        if children is not None:
            self.child(children)
    
    def child(self, children):
        for i in children:
            self.children.append(i)
        
    def __str__(self):
        return "...".join([str(i.value) for i in self.children])
            
    def go_deep(self, root):
        if root.children == []:
            # print("leaf", root.value)
            return root.value
        if root.children != []:
            summa = 0
            for i in root.children:   
                summa += self.go_deep(i)
        if summa < root.value:
            
            return root.value
        return summa
    
                

amount = int(input())
rinteet = [Tree() for x in range(amount)]
for i in range(amount-1):
    edge = input().split()
    rinteet[int(edge[0])-1].child([rinteet[int(edge[1])-1]])
values = input().split()
# print(values)
for i, v in enumerate(values):
    rinteet[i].value = int(v)
# for i in rinteet:
#     print(i.value, i)
print(rinteet[0].go_deep(rinteet[0]))
    



# t = Tree(0)
# t.child([Tree(2), Tree(4, [Tree(3, [Tree(3), Tree(4)]), Tree(1)])])
# print(t.children)
# print(t.go_deep(t))

Test 1

Group: 1, 2

Verdict: ACCEPTED

Test 2

Group: 1, 2

Verdict: ACCEPTED

Test 3

Group: 1, 2

Verdict: ACCEPTED

Test 4

Group: 2

Verdict: ACCEPTED

Test 5

Group: 2

Verdict: ACCEPTED

Test 6

Group: 1, 2

Verdict: ACCEPTED

Test 7

Group: 2

Verdict: ACCEPTED

Test 8

Group: 1, 2

Verdict: ACCEPTED

Test 9

Group: 2

Verdict: RUNTIME ERROR

Traceback (most recent call last):
  File "/box/input/code.py", line 43, in <module>
    p...

Test 10

Group: 1, 2

Verdict: ACCEPTED

Test 11

Group: 2

Verdict: ACCEPTED

Test 12

Group: 1, 2

Verdict: ACCEPTED

Test 13

Group: 2

Verdict: ACCEPTED