Laskettelukeskus

#Tässä ongelmana se, että jos on monta rinnettä, jotka lähtee samasta rinteestä. Pitää viel testaa toimiiko monet yksittäiset peräkkäiset rinteet
#otetaan syöte vastaan ja lisätään listoihin a, b ja c
a = [int(x) for x in input().split()]
b=[]
c=[]
for i in range(1,a[0]):
    e = [int(x) for x in input().split()]
    #print("e on:",e)
    for i in e:
        #print("i on:",i)
        b.append(i)
        #print("b on:",b)
e = [int(x) for x in input().split()]
for i in e:
    c.append(i) 
#print(b)
#print(c)
l=0
lauseke=0
d=0
valmis=False
#tehdään looppi, joka jatkuu kunnes vikassa listassa on vain 1 arvo
while len(c)>1:
    #tsekataan montako [1,2,1,3,3,4,3,5] [5,2,2,0,1] [5]
    while valmis!=True:
        #print("l",l)
        #print("1",b[(-2-(2*l))])
        #print("2",b[-4-(2*l)])
        if b[-2-2*l]==b[-4-2*l]:
            l+=1
        else:
            valmis=True
        if l==a[0]-2:
            valmis=True
    if l>=1:
        #print(b)
        #print(c)
        lauseke=c[b[-1]-1]
        d=3
        for i in range(1,l+1):
            lauseke+=c[b[-d]-1]
            d+=2
        if lauseke<=c[b[-2]-1]:
            for i in range(0,4):
                #print(b)
                b.pop()
            for i in range(1,l+2):
                c.pop()
        else:
            for i in range(1,3+2*l):
                b.pop()
            for i in range(1,l+2):
                c.pop()
            c[-1]=lauseke
    else:
        if c[-1]<=c[-2]:
            c.pop()
            b.pop()
            b.pop()
        else:
            c[-2]=c[-1]
            c.pop()
            b.pop()
            b.pop()
    #print("jep")
print(c[0])    

Test 1

Group: 1, 2

Verdict: ACCEPTED

Test 2

Group: 1, 2

Verdict: WRONG ANSWER

Test 3

Group: 1, 2

Verdict: WRONG ANSWER

Test 4

Group: 2

Verdict: WRONG ANSWER

Test 5

Group: 2

Verdict: WRONG ANSWER

Test 6

Group: 1, 2

Verdict: ACCEPTED

Test 7

Group: 2

Verdict: ACCEPTED

Test 8

Group: 1, 2

Verdict: ACCEPTED

Test 9

Group: 2

Verdict: ACCEPTED

Test 10

Group: 1, 2

Verdict: WRONG ANSWER

Test 11

Group: 2

Verdict: WRONG ANSWER

Test 12

Group: 1, 2

Verdict: WRONG ANSWER

Test 13

Group: 2

Verdict: WRONG ANSWER