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
 
 
def stupid():
    amount = int(input())
    rinteet = {key:[] for key in range(1, amount+1)}
    for i in range(amount-1):
        edge = input().split()
        rinteet[int(edge[0])].append(edge[1])
    leafs = [i[0] for i in rinteet.items() if i[1]==[]]
    values = input().split()
    summa = 0
    for i in leafs:
        summa+=int(values[i-1])
    # print(leafs)
    print(summa)
           
stupid()     
 
# 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: WRONG ANSWER

Test 3

Group: 1, 2

Verdict: WRONG ANSWER

Test 4

Group: 2

Verdict: WRONG ANSWER

Test 5

Group: 2

Verdict: WRONG ANSWER

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: ACCEPTED

Test 10

Group: 1, 2

Verdict: WRONG ANSWER

Test 11

Group: 2

Verdict: WRONG ANSWER

Test 12

Group: 1, 2

Verdict: ACCEPTED

Test 13

Group: 2

Verdict: ACCEPTED