Coding Test/이것이 코딩 테스트다
[이코테 다이나믹 프로그래밍] 못생긴 수
Lucian_Cho
2021. 1. 10. 05:26
# 문제
못생긴 수란 오직 2, 3, 5만을 소인수(Prime Factor)로 가지는 수를 의미합니다. 다시 말해 오직 2, 3, 5를 약수로 가지는 합성수를 의미합니다. 1은 못생긴 수라고 가정합니다. 따라서 못생긴 수들은 {1, 2, 3, 4, 5, 6, 8, 9, 10, 12, 15, ...} 순으로 이어지게 됩니다. 이때, n번째 못생긴 수를 찾는 프로그램을 작성하세요. 예를 들어 11번째 못생긴 수는 15입니다.
# 입력
- 첫째 줄에 n이 입력됩니다. (1 ≤ n ≤ 1,000)
# 출력
- n번째 못생긴 수를 출력합니다.
# 입력 예시
# 입력 예시 1
10
# 입력 예시 2
4
# 출력 예시
# 출력 예시 1
12
# 출력 예시 2
4
내 풀이
n = int(input())
dp = [0] * (n + 1) # dp 테이블 생성
dp[1] = 1
# 다이나믹 프로그래밍 진행 (바텀업)
for i in range(2, n + 1):
# 이전 못생긴 수 참조
j = dp[i - 1]
# 이전 못생긴 수의 다음 수부터 1개씩 차례대로 못생긴 수 탐색
while True:
j += 1
j_tmp = j
check = True # 못생긴 수가 맞는지 체크
while j_tmp != 1:
if j_tmp % 2 == 0:
j_tmp //= 2
elif j_tmp % 3 == 0:
j_tmp //= 3
elif j_tmp % 5 == 0:
j_tmp //= 5
else:
check = False
break
# 못생긴 수를 찾으면 i번째 못생긴 수로 기록
if check:
dp[i] = j
break
print(dp[n])
모범 답안 - 아직 이해가 좀 안된다...
n = int(input())
ugly = [0] * n # 못생긴 수를 담기 위한 테이블 (1차원 DP 테이블)
ugly[0] = 1 # 첫 번째 못생긴 수는 1
# 2배, 3배, 5배를 위한 인덱스
i2 = i3 = i5 = 0
# 처음에 곱셈 값을 초기화
next2, next3, next5 = 2, 3, 5
# 1부터 n까지의 못생긴 수들을 찾기
for l in range(1, n):
# 가능한 곱셈 결과 중에서 가장 작은 수를 선택
ugly[l] = min(next2, next3, next5)
# 인덱스에 따라서 곱셈 결과를 증가
if ugly[l] == next2:
i2 += 1
next2 = ugly[i2] * 2
if ugly[l] == next3:
i3 += 1
next3 = ugly[i3] * 3
if ugly[l] == next5:
i5 += 1
next5 = ugly[i5] * 5
# n번째 못생긴 수를 출력
print(ugly[n - 1])
문제점
시간 복잡도는 문제 없을 것 같은데, 코드 depth가 깊다.