반응형
문제
https://www.acmicpc.net/problem/2331
다음과 같이 정의된 수열이 있다.
- D[1] = A
- D[n] = D[n-1]의 각 자리의 숫자를 P번 곱한 수들의 합
예를 들어 A=57, P=2일 때, 수열 D는 [57, 74(=52+72=25+49), 65, 61, 37, 58, 89, 145, 42, 20, 4, 16, 37, …]이 된다. 그 뒤에는 앞서 나온 수들(57부터가 아니라 58부터)이 반복된다.
이와 같은 수열을 계속 구하다 보면 언젠가 이와 같은 반복수열이 된다. 이때, 반복되는 부분을 제외했을 때, 수열에 남게 되는 수들의 개수를 구하는 프로그램을 작성하시오. 위의 예에서는 [57, 74, 65, 61]의 네 개의 수가 남게 된다.
입력
첫째 줄에 A(1 ≤ A ≤ 9999), P(1 ≤ P ≤ 5)가 주어진다.
출력
첫째 줄에 반복되는 부분을 제외했을 때, 수열에 남게 되는 수들의 개수를 출력한다.
코드
import sys
input = sys.stdin.readline
a, p=input().split()
A=[]
A.append(int(a))
cnt=1
while True:
num, i=0, 0
while i<len(a):
num+=int(a[i])**int(p)
i+=1
a=(str(num))
if num in A:
print(A.index(num))
break
else:
A.append(num)
문제 해결
수학 구현 문제이다. 주어진 조건대로 하다보면 PS가 가능하다
반응형
'Problem Solving > 백준' 카테고리의 다른 글
[백준] 2407번: 조합 - [Python/파이썬] (0) | 2023.04.25 |
---|---|
[백준] 2338번: 긴자리 계산 - [Python/파이썬] (0) | 2023.04.25 |
[백준] 2295번: 세 수의 합 - [Python/파이썬] (0) | 2023.04.25 |
[백준] 2225번: 합분해 - [Python/파이썬] (0) | 2023.04.25 |
[백준] 2193번: 이친수 - [Python/파이썬] (0) | 2023.04.15 |