[백준/구현/C++] 2331번 반복수열

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

2331번: 반복수열

1. 우선 수를 한자리씩 분리한다
참고 : https://develop-me-z.tistory.com/103

[C++] 숫자 한자리씩 분리하기

2. 각 자리를 거듭제곱하면서 더해준다
참고 : https://develop-me-z.tistory.com/104

[C++] 수 거듭 제곱하기 (POW 함수)

3. 벡터에 구한 수가 들어있는지 확인하고 없으면 벡터에 넣고 있으면 인덱스를 반환한다
참고 : https://develop-me-z.tistory.com/105

[C++] vector - find 함수 (값의 존재 유무 확인 & 위치 구하기)

#include <bits/stdc++.h>

using namespace std;

int a, p;
vector<int> v;
bool tmp;

int main(void) {
    cin >> a >> p;

    v.push_back(a);

    int sum;
    int result = 0;
    while(tmp != true){
        sum = 0;
        while(a!=0){
            int rest = a%10;
            sum += pow(rest, p);
            a /= 10;
        }
        if(find(v.begin(), v.end(), sum)==v.end()){ // v에 없으면
            v.push_back(sum);
        }else{
            tmp = true;
            result = find(v.begin(), v.end(), sum) - v.begin(); // 인덱스 출력
        }
        a = sum;
    }
    cout << result << endl;

    return 0;
}