BOJ)2331 반복수열

문제 : https://www.acmicpc.net/problem/2331

문제의 INPUT을 배열해보면 위의 그림과 같은 형식으로 나오게 된다.

다음 숫자(next number)를 구하는 방법은 

while (A != 0) { calcul_res += (int)pow(A % 10, P); A /= 10; }

을 이용하면 되고,

백터의 처음 부터 끝까지를 탐색하며, 다음 숫자와 동일한 경우 함수를 끝마친다.

그렇지 않을 경우, 백터에 지속적으로 다음 숫자를 추가해준다.

#include <iostream> #include <vector> #include <math.h> using namespace std; int A, P; int calcul_res = 0; int res = 0, position = 0; vector<int> arr; int function() { arr.push_back(A); while (1) { while (A != 0) { calcul_res += (int)pow(A % 10, P); A /= 10; } for (int i = 0; i < arr.size(); i++) { if (calcul_res == arr[i]) { return i; } } arr.push_back(calcul_res); A = calcul_res; calcul_res = 0; } } int main() { cin >> A >> P; cout << function() << "\n"; return 0; }