[백준] 2331번 반복수열

💡문제접근

• 반복되는 값이 리스트 안에 존재한다면 break를 실행시켜 멈춘다.
• 그런 다음 반복되는 값이 나타나는 인덱스를 찾아 반복되는 부분을 제거해준다.

💡코드(메모리 : 30616KB, 시간 : 36ms)

A, P = map(int, input().split())

D = [A]
while True:
    total = 0
    for i in str(D[-1]):
        total += int(i) ** P

    if total in D:
        break
    D.append(total)

D = D[:D.index(total)]
print(len(D))

💡소요시간 : 4m