문제
두 자연수 N과 P를 가지고 다음 과정을 거쳐서 나오는 수를 차례대로 출력해보자.
처음 출력하는 수는 N이고, 두 번째 이후 출력하는 수는 N을 곱하고 P로 나눈 나머지를 구하는 과정을 반복하여 구한다.
즉, 먼저 N에 N을 곱하고, 이 수를 P로 나눈 나머지를 두 번째에 출력한다.
다음에는 이 나머지에 N을 곱하고 P로 나눈 나머지를 출력한다.
다음에는 이 나머지에 N을 곱한 후 P로 나눈 나머지를 출력한다.
이 과정을 계속 반복해보면 출력되는 에는 반복되는 부분이 있다.
예를 들어서, N = 67, P = 31인 경우를 생각해보자.
처음 출력되는 수는 67이고, 두 번째로 출력되는 수는 67×67 = 4489를 31로 나눈 나머지 25이다.
다음에는 25×67 = 1675를 31로 나눈 나머지 1, 다음에는 1×67 = 67을 31로 나눈 나머지 5가 차례대로 출력된다.
다음에는 5×67 = 335를 31로 나눈 나머지 25가 출력되는데, 이 수는 이미 이전에 출력된 수이다.
이 과정을 그림으로 보이면 다음과 같다.
즉 이 과정을 반복하면, 처음 67을 제외하면 3개의 수 25, 1, 5가 계속 무한히 반복되게 된다.
또 다른 예로, N = 9, P = 3을 가지고 시작하면, 9×9 = 81이고 3으로 나눈 나머지는 0이며, 0×3 = 0이고 3으로 나눈 나머지도 0이기 때문에 처음 9를 제외하면 0이 무한히 반복되게 된다.
N과 P를 입력받아 위와 같이 정의된 연산을 수행하였을 때, 반복되는 부분에 포함된 서로 다른 수의 개수를 구하는 프로그램을 작성하시오.
https://www.acmicpc.net/problem/2526
입력
첫째 줄에 처음 시작하는 두 자연수 N과 P가 공백을 사이에 두고 주어진다.
출력
첫째 줄에 반복되는 부분에 포함된 서로 다른 수의 개수를 출력한다.
예제
# input
67 31
# output
3
# input
96 61
# output
60
문제 풀이
# 입력 받기
N, P = map(int, input().split())
# 나머지들을 저장할 배열 초기화
arr = []
# 초기 나머지 값은 N
rest = N
while True:
# 다음 나머지 값 계산
rest = rest * N % P
# 계산한 나머지 값이 배열에 존재하면 break
if rest in arr:
break
# 나머지 값을 배열에 추가
arr.append(rest)
# 주기가 시작된 지점 이후의 길이를 계산하여 출력
print(len(arr)-arr.index(rest))
파이썬을 독학하시는 분들에게 도움이 되길 바라며,
혹 더 좋은 방법이 있거나 오류가 있다면 편하게 말씀 부탁드립니다.