[백준] 2331번 문제 (반복수열) 파이썬(Python) 풀이

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

2331번: 반복수열

n, m = map(int, input().split())
cur_num = n
nums = [cur_num]

while True:
    temp = str(cur_num)
    num = 0

    for i in temp:
        num += int(i) ** m

    if num in nums:
        nums = nums[:nums.index(num)]
        break
    else:
        nums.append(num)
        cur_num = num

print(len(nums))

딱히 설명할 게 없다..

문제에서 요구하는 대로 구현했다.