https://www.acmicpc.net/problem/2331
#include<stdio.h>
#include<math.h>
static int make(int a,int p){
int sum=0;
while(a>0){
sum+=pow(a%10,p);
a/=10;
}
return sum;
}
int main() {
int arr[100000];
int a,p;
scanf("%d %d",&a,&p);
arr[0]=a;
int c=0;
for(int i=1;;i++){
int m=arr[i-1];
int num=make(m,p);
for(int j=0;j<i;j++){
if(arr[j]==num){
printf("%d",j);
c=1;
break;
}
}
if(c==1) break;
arr[i]=num;
}
}