Baekjoon 1149
in Study / Computer science on Baekjoon 1149
dynamic programming
Description
RGB거리에는 집이 N개 있다. 거리는 선분으로 나타낼 수 있고, 1번 집부터 N번 집이 순서대로 있다.
집은 빨강, 초록, 파랑 중 하나의 색으로 칠해야 한다. 각각의 집을 빨강, 초록, 파랑으로 칠하는 비용이 주어졌을 때, 아래 규칙을 만족하면서 모든 집을 칠하는 비용의 최솟값을 구해보자.
Condition
- 1번 집의 색은 2번 집의 색과 같지 않아야 한다.
- N번 집의 색은 N-1번 집의 색과 같지 않아야 한다.
- i(2 ≤ i ≤ N-1)번 집의 색은 i-1번, i+1번 집의 색과 같지 않아야 한다.
Input
첫째 줄에 집의 수 N(2 ≤ N ≤ 1,000)이 주어진다. 둘째 줄부터 N개의 줄에는 각 집을 빨강, 초록, 파랑으로 칠하는 비용이 1번 집부터 한 줄에 하나씩 주어진다. 집을 칠하는 비용은 1,000보다 작거나 같은 자연수이다.
Output
첫째 줄에 모든 집을 칠하는 비용의 최솟값을 출력한다.
Example I & O
Input
3
26 40 83
49 60 57
13 89 99
Output 96
My solution
house_n = int(input())
dp = [[0]*3 for i in range(house_n+1)]
h_price = []
for i in range(house_n):
R, G, B = map(int, input().split())
h_price.append([R, G, B])
for i in range(1, len(dp)):
dp[i][0] = min(dp[i-1][1], dp[i-1][2]) + h_price[i-1][0]
dp[i][1] = min(dp[i-1][0], dp[i-1][2]) + h_price[i-1][1]
dp[i][2] = min(dp[i-1][0], dp[i-1][1]) + h_price[i-1][2]
print(min(dp[house_n]))
- Sketch
1, 2, 3 : house
DP table : 2D array for price → must be filled completely
ex) 2&R = min(40, 83) + 49