Laskettelukeskus

from collections import Counter
n = input()
lista = []
for item in range(int(n)-1):
    lista.append(input().split(" "))
n2 = input().split(" ")
lista = [tuple(l) for l in lista]
nlist = []
aurat = 0
nlist = [item for sublist in lista for item in sublist]
a = dict(Counter(nlist))
while len(a) >= int(1):
    dellist = []
    for item in a:
        if a[item] == 1:
            m = 0
            while True:
                m += 1
                #print((str(int(item)-m), (item)))
                if item == "1":
                    print(str(max(aurat, int(n2[0]))))
                    exit()
                if (str(int(item)-m), (item)) in lista:
                    dellist.append((item, str(int(item)-m)))
                    tarve = int(n2[int(item)-1])
                    aurat += tarve
                    h = int(item) - m - 1
                    n2[h] = str(max(int(n2[h]) - tarve, 0))
                    break
    #lista = [e for e in lista if e not in dellist]
    for item in dellist:
        a.pop(item[0])
        a[item[1]] -= 1
        #print(str(a))
print(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: ACCEPTED

Test 9

Group: 2

Verdict: ACCEPTED

Test 10

Group: 1, 2

Verdict: TIME LIMIT EXCEEDED

Test 11

Group: 2

Verdict: TIME LIMIT EXCEEDED

Test 12

Group: 1, 2

Verdict: TIME LIMIT EXCEEDED

Test 13

Group: 2

Verdict: TIME LIMIT EXCEEDED