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()))
juurenekaarvo=times[0]
def go_up(solmu):
    solmuarvo=times[solmu-1]
    nykysolmu=solmu
    times[solmu-1]=0
    juuriarvo=times[0]
    while nykysolmu!=1:
        nykysolmu=up[nykysolmu][0]
        times[nykysolmu-1]-=solmuarvo 
solmulista=[i for i in range(2,n+1)]
zerolist=[0 for i in range(n-1)]
while solmulista!=zerolist:
    for i in solmulista:
        if down[i]==[] and i!=0:
            if times[i-1]>0:
                go_up(i)
            solmulista[solmulista.index(i)]=0
            for j in range(1,n+1):
                if i in down[j]:
                    down[j].remove(i)
print(abs(times[0])+juurenekaarvo)

Test 1

Group: 1, 2

Verdict: ACCEPTED

Test 2

Group: 1, 2

Verdict: ACCEPTED

Test 3

Group: 1, 2

Verdict: ACCEPTED

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

Test 9

Group: 2

Verdict: TIME LIMIT EXCEEDED

Test 10

Group: 1, 2

Verdict: ACCEPTED

Test 11

Group: 2

Verdict: TIME LIMIT EXCEEDED

Test 12

Group: 1, 2

Verdict: ACCEPTED

Test 13

Group: 2

Verdict: TIME LIMIT EXCEEDED