[백준/파이썬] 2526번 싸이클

-Code

n, p = map(int, input().split())
num = n
arr = []
while 1:
    num = num * n % p
    if num in arr:
        print(len(arr) - arr.index(num))
        break
    arr.append(num)

처음에 틀려서 문제가 뭔지 찾아보니 입력이 2, 32가 들어가면 1이나 와야 하는데 4가 나오게 되어서 오류가 나서 index를 넣어서 해결해 주었습니다.