Laskettelukeskus

n = int(input())
 
slopes = {}
for i in range(n-1):
    slopes[i] = [int(x) for x in input().split()]
 
snowamount = [int(x) for x in input().split()]
 
amount = 0
 
def find_pair(num, daddy_number):
    global amount
    pairs = []
    
 
    
    for i in range(n-1):
        if(slopes[i][0] == num):
            pairs.append(slopes[i][1])
    number = 0
    if(len(pairs) > 0):
        for i in range(len(pairs)):
            number += snowamount[pairs[i]-1]
            find_pair(pairs[i], snowamount[pairs[i]-1])
    if(number < daddy_number):
        amount = amount + (daddy_number - number)
            
    if(len(pairs) == 0):
        amount = amount + daddy_number
 
 
for i in range(n-1):
    if(slopes[i][0] == 1):
        num = slopes[i][1]
        find_pair(slopes[i][1], snowamount[0])
      
 
 
print(amount)
 

Test 1

Group: 1, 2

Verdict: WRONG ANSWER

Test 2

Group: 1, 2

Verdict: WRONG ANSWER

Test 3

Group: 1, 2

Verdict: WRONG ANSWER

Test 4

Group: 2

Verdict: TIME LIMIT EXCEEDED

Test 5

Group: 2

Verdict: TIME LIMIT EXCEEDED

Test 6

Group: 1, 2

Verdict: ACCEPTED

Test 7

Group: 2

Verdict: TIME LIMIT EXCEEDED

Test 8

Group: 1, 2

Verdict: WRONG ANSWER

Test 9

Group: 2

Verdict: TIME LIMIT EXCEEDED

Test 10

Group: 1, 2

Verdict: WRONG ANSWER

Test 11

Group: 2

Verdict: TIME LIMIT EXCEEDED

Test 12

Group: 1, 2

Verdict: WRONG ANSWER

Test 13

Group: 2

Verdict: TIME LIMIT EXCEEDED