백준 싸이클 c++

https://www.acmicpc.net/problem/2526

2526번: 싸이클

문제

두 자연수 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를 입력받아 위와 같이 정의된 연산을 수행하였을 때,  반복되는 부분에 포함된 서로 다른 수의 개수를 구하는 프로그램을 작성하시오.

입력

첫째 줄에 처음 시작하는 두 자연수 N과 P가 공백을 사이에 두고 주어진다.

출력

첫째 줄에 반복되는 부분에 포함된 서로 다른 수의 개수를 출력한다.

제한

• 1 ≤ N ≤ 1,000
• 2 ≤ P ≤ 97

예제 입력 1 복사

67 31

예제 출력 1 복사

3

예제 입력 2 복사

96 61

예제 출력 2 복사

60

 

 

#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;
 
int N, P;
vector<int> v;
void InputData(){
    cin >> N >> P;
}
 
int sum(int num) {
    return (num * N)%P;
}
 
int solve(int n) {
    v.push_back(n);
    int idx = 0;
    while (true)
    {
        n = sum(n);
        if (find(v.begin(), v.end(), n) != v.end()) {
            idx = find(v.begin(), v.end(), n) - v.begin();
            break;
        }
        v.push_back(n);
    }
 
    return v.size() - idx;
     
}
 
int main(){
    int ans = -1;
    InputData();// 입력받는 부분
 
    // 여기서부터 작성
    ans = solve(N);
 
    cout << ans << endl;// 출력하는 부분
    return 0;
}