[백준/Python3] 17404 RGB거리2

Problem🧡💚💙

https://www.acmicpc.net/problem/17404
RGB거리에는 집이 N개 있다. 거리는 선분으로 나타낼 수 있고, 1번 집부터 N번 집이 순서대로 있다.

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

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

Solution

각 집은 양 옆의 집과 다른 색을 칠해야 하고 첫 번째 집은 마지막 집과 다른 색이어야 한다. 이 조건을 만족하기 위해서 먼저 첫 번째 집의 색깔을 정하고 집을 칠하기 시작한다. 마지막 집까지 칠한 뒤에는 첫 번째 집과 다른 색을 칠한 경우 중 가장 최소 비용이 드는 경우를 찾으면 된다.

Python Code

from sys import stdin

N = int(stdin.readline())
R, G, B = 0, 1, 2
cost = [list(map(int, stdin.readline().split())) for _ in range(N)]
INF = float("inf")
ans = INF
for color in [R, G, B]:
    dp = [[INF, INF, INF] for _ in range(N)]
    dp[0][color] = cost[0][color]
    for i in range(1, N):
        dp[i][R] = cost[i][R] + min(dp[i - 1][G], dp[i - 1][B])
        dp[i][G] = cost[i][G] + min(dp[i - 1][R], dp[i - 1][B])
        dp[i][B] = cost[i][B] + min(dp[i - 1][R], dp[i - 1][G])
    dp[-1][color] = INF
    ans = min(ans, mint(dp[-1]))
print(ans)