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()
#define MOD 1000000009

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 tmp = 0;

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

int     main()
{
    int n, p;
    cin >> n >> p;

    vector<ii> count(240000);

    for (int i = 0;count[n].xx != 1; i++)
    {
        count[n].xx++;
        count[n].yy = i;
        n = calc(n, p);
    }
    cout << count[n].yy << endl;

    return 0;
}