Laskettelukeskus

import sys
from collections import Counter
n = sys.stdin.readline()
lista = []
for item in range(int(n)-1):
    lista.append(input().split(" "))
n2 = sys.stdin.readline().split(" ")
    
d = {}
nlist = []
aurat = 0
while len(lista) > int(1):
    nlist = [item for sublist in lista for item in sublist]
    a = Counter(nlist)
    dellist = []
    for item in a:
        if a[item] == 1:
            m = 0
            while True:
                m += 1
                if [str(int(item)-m), (item)] in lista:
                    dellist.append([str(int(item)-m), (item)])
                    tarve = int(n2[int(item)-1])
                    aurat += tarve
                    h = int(item) - m - 1
                    n2[h] = str(max(int(n2[h]) - tarve, 0))
                    break
    for item in dellist:
        lista.remove(item)
sys.stdout.write(str(max(aurat, int(n2[0]))))

Test 1

Group: 1, 2

Verdict: ACCEPTED

Test 2

Group: 1, 2

Verdict: TIME LIMIT EXCEEDED

Test 3

Group: 1, 2

Verdict: TIME LIMIT EXCEEDED

Test 4

Group: 2

Verdict: TIME LIMIT EXCEEDED

Test 5

Group: 2

Verdict: TIME LIMIT EXCEEDED

Test 6

Group: 1, 2

Verdict: TIME LIMIT EXCEEDED

Test 7

Group: 2

Verdict: TIME LIMIT EXCEEDED

Test 8

Group: 1, 2

Verdict: TIME LIMIT EXCEEDED

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