다음은 동적 계획법을 활용한 예시이다.
def fibonacci(n):
if n <= 1:
return n
# 작은 문제의 정답을 저장할 배열 초기화
dp = [0] * (n+1)
dp[1] = 1
dp[2] = 1
# 큰 문제를 작은 문제로 분할하여 풀기
for i in range(3, n+1):
dp[i] = dp[i-1] + dp[i-2]
# 큰 문제의 정답 반환
return dp[n]
정수 X에 사용할 수 있는 연산은 다음과 같이 세 가지이다.
정수 N이 주어졌을 때, 위와 같은 연산 세 개를 적절히 사용해서 1을 만들려고 한다. 연산을 사용하는 횟수의 최솟값을 출력하시오.
첫째 줄에 1보다 크거나 같고, 106보다 작거나 같은 정수 N이 주어진다.
첫째 줄에 연산을 하는 횟수의 최솟값을 출력한다.
10의 경우에 10 → 9 → 3 → 1 로 3번 만에 만들 수 있다.
전의 결과를 다음 결과에 이용하게 되는 점화식을 활용한 DP 문제이다.
⏩ 메모제이션 방법으로 중복해 계산되는 값을 저장해 효율을 높여준다.
처음 두 수를 알기 때문에 상향식(bottom-up)으로 풀 수 있다.
X = 10인 경우, 10 -> 9 -> 3 -> 1 과정을 거쳐 1이 되게 되는데
9의 경우에는 또, 9 -> 3 -> 1의 과정을 거치며
3의 경우에는 3 -> 1의 과정을 거친다.
즉, 10을 구할 때는 9의 결과를, 9를 구할 때는 3의 결과를 이용한다.
앞에서 구한 결과값을 저장하였다가 후에 사용하는 것이다.
일단, 2와 3으로 나누어 떨어지지 않는 경우는 무조건 1을 빼야 하기 때문에
dp[i] = dp[i-1] + 1을 통해 횟수를 +1 해준다.
그리고나서, dp[i]가 2 와 3으로 나누어 떨어지는 경우에는 dp[i](1을 빼는 경우)와 dp[i//2or3]+1(나누었을 경우) 중 최소값을 선택한다.
n = int(input())
# DP 테이블 초기화
d = [0] * 1000001
# 다이나믹 프로그래밍 진행(bottom-up)
for i in range(2, n+1):
# 현재의 수에서 1을 빼는 경우
d[i] = d[i-1] + 1
# 현재의 수가 2로 나누어 떨어지는 경우
if i%2 == 0:
d[i] = min(d[i], d[i//2] + 1)
# 현재의 수가 3으로 나누어 떨어지는 경우
if i%3 == 0:
d[i] = min(d[i], d[i//3] + 1)
# 결과 출력
print(d[n])
참고
[백준] 1463번 1로 만들기 (Python 파이썬)
[파이썬] 백준 - 1463: 1로 만들기 (매우 자세한 해설 포함)