[BOJ][⚪1][백준#01149] RGB거리

문제 출처

BAEKJOON Online Judge #1149

문제

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보다 작거나 같은 자연수이다.

출력

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

예제

예제 1

입력

3
26 40 83
49 60 57
13 89 99

출력

96

예제 2

입력

3
1 100 100
100 1 100
100 100 1

출력

3

예제 3

입력

3
1 100 100
100 100 100
1 100 100

출력

102

예제 4

입력

6
30 19 5
64 77 64
15 19 97
4 71 57
90 86 84
93 32 91

출력

208

예제 5

입력

8
71 39 44
32 83 55
51 37 63
89 29 100
83 58 11
65 13 15
47 25 29
60 66 19

출력

253

My Sol

import sys
input = sys.stdin.readline

N = int(input())
mat = [list(map(int, input().split())) for _ in range(N)]
dictC = {0:(1,2), 1:(0,2), 2:(0,1)}

i = 1
while i<N:
    for j in range(3):
        c1, c2 = dictC[j]
        v = min(mat[i-1][c1], mat[i-1][c2])
        mat[i][j] += v
    i += 1

print(min(mat[N-1]))

DP 방식과 비슷한 덮어씌우기 방식을 사용했다. mat으로 입력을 받고, i번째 집은 현재 색깔을 선택했을 때, 이 색깔을 직전에 사용하지 않은 경우 중 가장 적은 비용과 현재 집을 이 색깔로 했을 때의 비용을 더한 값을 현재 mat에 씌워준다. 이렇게 끝까지 가서, 각 색깔 중 최솟값을 출력하면 되겠다.

결과

맞았습니다!!

모범답안

출처

# empty