스터디/알고리즘

반복수열

세졍 2024. 3. 9. 13:22

 

 

1. 각 자릿수를 P번곱해서 더한수를 수열에 넣기때문에 수를 str으로 바꿀필요

2. 같은 수가 들어오면 그때부터는 반복이므로 같은수가 있다면 수열 넣지 않고 break

3. 같은수가 들어올 경우  수열안에 들어있었던 그수의 인덱스를 구하면 그 수가 반복수열을 제외하고 

수열에 남게 되는 수들의 개수이다

 

 

A , P = map(int,input().split())
lst = [A]
idx = 0
while True:
    tmp =0
    for i in str(lst[-1]):
        tmp += int(i)**P
    if tmp in lst:
        idx = lst.index(tmp)
        break
    lst.append(tmp)

print(idx)