Laskettelukeskus

from collections import Counter
import numpy
n = input()
lista = []
for item in range(int(n)-1):
    lista.append(input().split(" "))
n2 = input().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)
print(str(max(aurat, int(n2[0]))))

Test 1

Group: 1, 2

Verdict: RUNTIME ERROR

Traceback (most recent call last):
  File "input/code.py", line 2, in <module>
    import...

Test 2

Group: 1, 2

Verdict: RUNTIME ERROR

Traceback (most recent call last):
  File "input/code.py", line 2, in <module>
    import...

Test 3

Group: 1, 2

Verdict: RUNTIME ERROR

Traceback (most recent call last):
  File "input/code.py", line 2, in <module>
    import...

Test 4

Group: 2

Verdict: RUNTIME ERROR

Traceback (most recent call last):
  File "input/code.py", line 2, in <module>
    import...

Test 5

Group: 2

Verdict: RUNTIME ERROR

Traceback (most recent call last):
  File "input/code.py", line 2, in <module>
    import...

Test 6

Group: 1, 2

Verdict: RUNTIME ERROR

Traceback (most recent call last):
  File "input/code.py", line 2, in <module>
    import...

Test 7

Group: 2

Verdict: RUNTIME ERROR

Traceback (most recent call last):
  File "input/code.py", line 2, in <module>
    import...

Test 8

Group: 1, 2

Verdict: RUNTIME ERROR

Traceback (most recent call last):
  File "input/code.py", line 2, in <module>
    import...

Test 9

Group: 2

Verdict: RUNTIME ERROR

Traceback (most recent call last):
  File "input/code.py", line 2, in <module>
    import...

Test 10

Group: 1, 2

Verdict: RUNTIME ERROR

Traceback (most recent call last):
  File "input/code.py", line 2, in <module>
    import...

Test 11

Group: 2

Verdict: RUNTIME ERROR

Traceback (most recent call last):
  File "input/code.py", line 2, in <module>
    import...

Test 12

Group: 1, 2

Verdict: RUNTIME ERROR

Traceback (most recent call last):
  File "input/code.py", line 2, in <module>
    import...

Test 13

Group: 2

Verdict: RUNTIME ERROR

Traceback (most recent call last):
  File "input/code.py", line 2, in <module>
    import...