본문 바로가기
스터디/알고리즘

반복수열

by 세졍 2024. 3. 9.

 

 

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)

'스터디 > 알고리즘' 카테고리의 다른 글

카드2  (0) 2024.03.09
단지번호 붙이기  (0) 2024.03.09
동적계획법(DP)  (0) 2024.03.07
순열사이클  (0) 2024.03.07
타겟넘버  (1) 2024.02.28