백준 2331 반복수열

a, p = map(int, input().split())

d = []
d.append(a)
while True:
    s = str(d[-1])
    tmp = 0
    for x in s:
        tmp += (int(x) ** p)
    if tmp in d:
        break
    else:
        d.append(tmp)
for i in range(len(d)):
    if d[i] == tmp:
        print(i)

하라는 대로 ..