2331번: 반복수열

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


$O(...)$

#include<cstdio>
#include<cmath>
int ck[236197], a, p, cnt;
int main() {
    scanf("%d%d", &a, &p);
    for (; !ck[a];) {
        ck[a] = ++cnt;
        int s = 0;
        for (int i = a; i; i /= 10) s += pow(i % 10, p);
        a = s;
    }
    printf("%d", ck[a] - 1);
    return 0;
}