AtCoder Beginners Contest 065 B - Trained?について

Question

頭をどう動かしてもこれがWAになる理由が全くわからないので誰か教えてください・・・
問題

N = int(input())
A = [0]  # parser
A += [int(input()) for i in range(N)]
count = 1
flashed = A[1]
while True:
    flashed = A[flashed]
    count += 1
    if flashed == 2:
        print(count)
        exit(0)
    elif count > 10 ** 6:
        print(-1)
        exit(0)

Answer 1

問題文からすると、1回ボタン押しただけで終わることもあるはずです。ご提示のコードだと最低でも(count>=2) になってWAとなるのでは。（他の原因もあるかもしれませんが、少なくとも原因のひとつとして）

# ここで1になって
count = 1
flashed = A[1]
while True:
    flashed = A[flashed]
    # これで必ず2以上になってしまう
    count += 1

Comments

[reud]

解けました！初歩的ミスでしたね。。。。ありがとうございました！！