[Baekjoon]3331繰り返し数列python

🏷 質問する

💡 コード#コード#

a, p = map(int, input().split())

seq = []
seq.append(a)
num = a
while True:
    res = 0
    for i in str(num):
        res += int(i)**p
    if res in seq:
        idx = seq.index(res)
        break
    else:
        seq.append(res)
        num = res
        # print(seq)
print(idx)

🔑