Problem p02684 (python)

Fixed submission:
n, k = map(int,input().split())
a = list(map(int, input().split()))
t = [1]
x = [0]*n
x[0] = 1
y = 1
z = 0
d = 0
flag = False
for i in range(k):
    t.append(a[t[i]-1])
    if x[a[t[i]-1]-1] != 0:
        y = x[a[t[i]-1]-1]
        z = i+2
        b = [a[t[i]-1]]
        break
    else:
        x[a[t[i]-1]-1] += i+2
        z = i+2
        b = [a[t[i]-1]]
    if i == k - 1:
        flag = True

if flag:
    print(a[t[k-1]-1])
    exit()

d = z-y


if d != 0:
    r = (k - y) % d
else:
    r = k - y

for i in range(r+1):
    b.append(a[b[i]-1])

print(b[r+1])

Buggy submission (verdict: Runtime Error):
n, k = map(int,input().split())
a = list(map(int, input().split()))
t = [1]
x = [0]*n
x[0] = 1
y = 1
z = 0
d = 0
for i in range(k):
    t.append(a[t[i]-1])
    if x[a[t[i]-1]-1] != 0:
        y = x[a[t[i]-1]-1]
        z = i+2
        b = [a[t[i]-1]]
        break
    else:
        x[a[t[i]-1]-1] += i+2
        z = i+2

d = z-y


if d != 0:
    r = (k - y) % d
else:
    r = k - y

for i in range(r+1):
    b.append(a[b[i]-1])

print(b[r+1])

Buggy submission (verdict: Runtime Error):
n, k = map(int,input().split())
a = list(map(int, input().split()))
t = [1]
x = [0]*n
x[0] = 1
y = 1
z = 0
d = 0
for i in range(k):
    t.append(a[t[i]-1])
    if x[a[t[i]-1]-1] != 0:
        y = x[a[t[i]-1]-1]
        z = i+2
        b = [a[t[i]-1]]
        break
    else:
        x[a[t[i]-1]-1] += i+2
        z = i+2

d = z-y

print(t)
print(x)
print(y)
print(z)

if d != 0:
    r = (k - y) % d
else:
    r = k - y

for i in range(r+1):
    b.append(a[b[i]-1])

print(b[r+1])

Buggy submission (verdict: Runtime Error):
n, k = map(int,input().split())
a = list(map(int, input().split()))
t = [1]
x = [0]*n
x[0] = 1
y = 0
z = 0
d = 0
for i in range(k):
    t.append(a[t[i]-1])
    if x[a[t[i]-1]-1] != 0:
        y = x[a[t[i]-1]-1]
        z = i+2
        b = [a[t[i]-1]]
        break
    else:
        x[a[t[i]-1]-1] += i+2
        z = i+2

d = z-y

if d != 0:
    r = (k - y) % d
else:
    r = k - y

for i in range(r+1):
    b.append(a[b[i]-1])

print(b[r+1])

Buggy submission (verdict: Runtime Error):
n, k = map(int,input().split())
a = list(map(int, input().split()))
t = [1]
x = [0]*n
x[0] = 1
y = 0
z = 0
d = 0
for i in range(k):
    t.append(a[t[i]-1])
    if x[a[t[i]-1]-1] != 0:
        y = x[a[t[i]-1]-1]
        z = i+2
        b = [a[t[i]-1]]
        break
    else:
        x[a[t[i]-1]-1] += i+2

d = z-y

if d != 0:
    r = (k - y) % d
else:
    r = k - y

for i in range(r+1):
    b.append(a[b[i]-1])

print(b[r+1])

Buggy submission (verdict: Runtime Error):
n, k = map(int,input().split())
a = list(map(int, input().split()))
t = [1]
d = 1
for i in range(k):
    t.append(a[t[i]-1])

    if len(t) != len(set(t)):

        l = [m for m, x in enumerate(t) if x == t[i+1]]
        b = [t[i+1]]
        break

d = l[1] - l[0]
if d != 0:
    r = (k - (l[0]+1)) % d
else:
    r = (k - (l[0]+1))

for i in range(r+1):
    b.append(a[b[i]-1])

print(b[r+1])

Buggy submission (verdict: Runtime Error):
n, k = map(int,input().split())
a = list(map(int, input().split()))
t = [1]
d = 1
for i in range(k):
    t.append(a[t[i]-1])

    if len(t) != len(set(t)):

        l = [m for m, x in enumerate(t) if x == t[i+1]]
        b = [t[i+1]]
        break

d = l[1] - l[0]
r = (k - (l[0]+1)) % d

for i in range(r+1):
    b.append(a[b[i]-1])

print(b[r+1])

Buggy submission (verdict: Runtime Error):
n, k = map(int,input().split())
a = list(map(int, input().split()))
t = [1]
d = 1
for i in range(k):
    t.append(a[t[i]-1])

    if len(t) != len(set(t)):

        l = [m for m, x in enumerate(t) if x == t[i+1]]
        b = [t[i+1]]
        break

d = l[1] - l[0]
r = (k - (l[0]+1)) % d

for i in range(r+1):
    b.append(a[b[i]-1])

print(b[r+1])