[백준 파이썬] 1149 RGB거리

문제

RGB거리에는 집이 N개 있다. 거리는 선분으로 나타낼 수 있고, 1번 집부터 N번 집이 순서대로 있다.

집은 빨강, 초록, 파랑 중 하나의 색으로 칠해야 한다. 각각의 집을 빨강, 초록, 파랑으로 칠하는 비용이 주어졌을 때, 아래 규칙을 만족하면서 모든 집을 칠하는 비용의 최솟값을 구해보자.

• 1번 집의 색은 2번 집의 색과 같지 않아야 한다.
• N번 집의 색은 N-1번 집의 색과 같지 않아야 한다.
• i(2 ≤ i ≤ N-1)번 집의 색은 i-1번, i+1번 집의 색과 같지 않아야 한다.

입력

첫째 줄에 집의 수 N(2 ≤ N ≤ 1,000)이 주어진다. 둘째 줄부터 N개의 줄에는 각 집을 빨강, 초록, 파랑으로 칠하는 비용이 1번 집부터 한 줄에 하나씩 주어진다. 집을 칠하는 비용은 1,000보다 작거나 같은 자연수이다.

출력

첫째 줄에 모든 집을 칠하는 비용의 최솟값을 출력한다.

---

풀이

조건은 결국 중복된 색상을 연속으로 사용하지않는 것이다.
가장 위에 2줄의 최소값을 구하려면 두번째 줄이 R을 선택한 경우 2가지, G를 선택한 경우 2가지, B를 선택한경우 2가지 에서 최소값을
각각의 최소값을 구하면 그게 각각 R을 선택했을 경우 최소값, G을 선택했을 경우 최소값, B을 선택했을 경우 최소값이 된다.
해당 최소값들과 3번째 줄을 같은 방식으로 계산하면 1,2 줄과 3번 째 줄의 최소값이 계산된다. 이런 방식으로 마지막 R, G, B값까지 계산하면된다.

 

 

코드

n = int(input())
 
cost = []
 
dp = [[0]*3 for _ in range(n)]
for i in range(n):
  cost.append(list(map(int, input().split())))
 
dp[0][0] = cost[0][0]
dp[0][1] = cost[0][1]
dp[0][2] = cost[0][2]
 
for i in range(1, n):
  dp[i][0] = min(dp[i-1][1] + cost[i][0], dp[i-1][2] + cost[i][0])
  dp[i][1] = min(dp[i-1][0] + cost[i][1], dp[i-1][2] + cost[i][1])
  dp[i][2] = min(dp[i-1][0] + cost[i][2], dp[i-1][1] + cost[i][2]) 
 
print(min(dp[n-1][0], dp[n-1][1], dp[n-1][2]))