Laskettelukeskus

import time
def canGo(a,b, paths):
    for x in paths:
        if a in x and b in x:
            return True
    return False

def findWay(target, paths):
    current = 1
    fol = 2
    way = [1]
    nots = []
    i = 0
    while current != target:
        through = True
        for x in paths:
            #print(f"{x[0]} == {current} and {x[1]} == {fol}")
            if(x[0] == str(current) and x[1] == str(fol) and fol not in nots and fol not in way):
                way.append(fol)
                current = fol
                fol += 1
                #print(way)
                through = False
                break
        if(through):
            #print("helo")
            fol += 1
            if(fol > target):
                nots.append(way[len(way)-1])
                way.pop()
                #print(f"Nots: {nots}")
                current = way[len(way)-1]
                fol = current+1
        i += 1
        #time.sleep(1)
    return way

linesA = int(input())
slopes = {}
paths = []
for i in range(linesA-1):
    paths.append(input().split(" "))
times = input().split(" ")
for i in range(1,linesA+1):
    slopes.update({i:int(times[i-1])})
a = 0
b = 0
sValues = list(slopes.values())
aurat = 0
while sValues.count(0) != len(sValues):
    a = 0
    b = 0
    sValues = list(slopes.values())
    while sValues.count(0) != len(sValues):
        i = 0
        for x in sValues[::-1]:
            if x != 0:
                paikka = len(sValues)-i
                rinne = paikka
                #print(f"Rinne {rinne}, paikka: {paikka}")
                break
            i += 1
        break
        a += 1
    way = findWay(rinne, paths)
    for x in  way:
        if slopes.get(int(x)) != 0:
            slopes.update({int(x):slopes.get(int(x))-1})
        else:
            slopes.update({int(x):0})
    aurat += 1
aurat = aurat -1
print(aurat)

Test 1

Group: 1, 2

Verdict: ACCEPTED

Test 2

Group: 1, 2

Verdict: RUNTIME ERROR

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

Test 3

Group: 1, 2

Verdict: RUNTIME ERROR

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

Test 4

Group: 2

Verdict: TIME LIMIT EXCEEDED

Test 5

Group: 2

Verdict: TIME LIMIT EXCEEDED

Test 6

Group: 1, 2

Verdict: WRONG ANSWER

Test 7

Group: 2

Verdict: WRONG ANSWER

Test 8

Group: 1, 2

Verdict: ACCEPTED

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