listの参照ができない

Question

ある問題をといており、以下のコードを書きました。

n,k = map(int,input().split())
a = list(map(int, input().split()))

i = 0
memo = []
cnt = 0
cnt2 = 0
s = 0
while True:
    memo.append(a[i])
    i = a[i]
    if i in memo[:-1]:
        cnt2 = memo[:-1].index(i)
        cnt = len(memo[cnt2:])
        cntlist = memo[cnt2]
        break
s = (k - cnt2)%cnt
ans = cntlist[s]

print(ans)

実行し、入力を与えると
以下のエラーがでました。

Traceback (most recent call last):
  File "d.py", line 11, in <module>
    memo.append(a[i])
IndexError: list index out of range

print([a[i])とすると出力されるので、リスト「a」の参照時にエラーが起きているわけではないとおもったのですが、、
どこでミスがおきているのでしょうか。
よろしくお願いします。

Comments

[meg_]

memo.append(a[i])の前にprint(a[i])を入れたらどうなりますか？

[mai-mo_n.]

4 5
3 2 4 1
という入力を与えると
3
3
1
2
4
Traceback (most recent call last):
File "d.py", line 11, in <module>
print(a[i])
IndexError: list index out of range
と、ある程度の出力の後、エラーがでました。
while文の外でしかprintの出力について調べていませんでした。。。

[mai-mo_n.]

リストの範囲からiが外れてしまうのは、シンプルに考え方の間違いからでした。。
申し訳ないです。
対応していただき、ありがとうございました。

Answer 1

リスト「a」の参照時にエラーが起きているわけではないとおもったのですが、、

リストaの添え字での参照時にエラーが起きていますので、この行の前にiの範囲チェックを入れましょう。範囲外だったらどうしたいのでしょう？

Comments

[mai-mo_n.]

print(i)を直前に入れて
4 5
3 2 4 1
という入力をあたえると、
0
3
1
2
4
以下のようなエラーがでました。
Traceback (most recent call last):
File "d.py", line 11, in <module>
memo.append(a[i])
IndexError: list index out of range
iがあきらかに範囲からはずれていました。
なぜ問題設定にあわずはずれてしまうかは、解決いたしました。
その問題に対する考え方をそもそも間違えていたからでした。
丁寧に対応していただき、ありがとうございました。