Laskettelukeskus

n=int(input())
up=[[]for i in range(n+1)]
down=[[]for i in range(n+1)]
for i in range(n-1):
    a,b=map(int,input().split())
    down[a].append(b)
    up[b].append(a)
times=list(map(int,input().split()))
def go_up(solmu):
    solmuarvo=times[solmu-1]
    nykysolmu=solmu
    times[solmu-1]=0
    juuriarvo=times[0]
    while times[0]==juuriarvo:
        nykysolmu=up[nykysolmu][0]
        times[nykysolmu-1]-=solmuarvo 
solmulista=[i for i in range(2,n-1)]
while solmulista!=[]:
    for i in solmulista:
        if down[1]==[]:
            go_up(i)
            solmulista.remove(i)
print(abs(times[0]))

Test 1

Group: 1, 2

Verdict: TIME LIMIT EXCEEDED

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