2331 반복수열

전에 풀었던 문제..

 

burningjeong.tistory.com/434

2331 반복수열

 

숫자가 중복돼서 나오면 그 수는 반복수열의 시작점이라는 소리니깐 이 위치를 출력하면 된다. 

 

 

#include <iostream>
#include <algorithm>
#include <vector>
#include <string>
#include <functional>
#include <string>
#include <queue>
#include <stack>
#include <set>
#include <map>
#include <cmath>
#include <cstring>
#include <bitset>

#define xx first
#define yy second
#define all(x) (x).begin(), (x).end()

using namespace std;
using i64 = long long int;
using ii = pair<int, int>;
using ii64 = pair<i64, i64>;
using iii = tuple<int, int, int>;

int     calc(int n, int p)
{
    int ans = 0;

    while (n > 0)
    {
        ans += pow(n % 10, p);
        n /= 10;
    }

    return ans;
}

int     main()
{
    int n, p;
    scanf("%d %d", &n, &p);

    vector<ii> count(240000);
    count[n].xx = 1;
    count[n].yy = 0;

    for (int i = 1; ; i++)
    {
        n = calc(n, p);

        if (count[n].xx == 1)
        {
            printf("%d\n", count[n].yy);
            return 0;
        }

        count[n].xx = 1;
        count[n].yy = i;
    }
    
    return 0;
}