📚 2331 - 반복 수열

반복수열

풀이

• 반복 수열이 되는 위치를 찾아야 한다. (그 위치전 인덱스가 마지막으로 반복되지 않는 구간이 된다.)
• 반복된 구간이 보일 때, 배열에 넣지 않고 나간 후, 처음으로 이 반복된 값이 들어온 인덱스를 출력해준다.(반복 시작점의 인덱스가 반복되는 부분을 제외했을 때의 인덱스이다.)

소스

from sys import stdin as s

a, p = map(int, s.readline().split())


cur_value = [a]

while True:
    cnt = 0

    for cur_data in str(cur_value[-1]):
        cnt += int(cur_data) ** p

    if cnt in cur_value:
        break

    cur_value.append(cnt)

print(cur_value.index(cnt))

채점 결과

참고

• https://leeiopd.tistory.com/entry/BOJ-2331-%EB%B0%98%EB%B3%B5%EC%88%98%EC%97%B4-Python-%ED%8C%8C%EC%9D%B4%EC%8D%AC