문제
RGB거리에는 집이 N개 있다. 거리는 선분으로 나타낼 수 있고, 1번 집부터 N번 집이 순서대로 있다.
집은 빨강, 초록, 파랑 중 하나의 색으로 칠해야 한다. 각각의 집을 빨강, 초록, 파랑으로 칠하는 비용이 주어졌을 때, 아래 규칙을 만족하면서 모든 집을 칠하는 비용의 최솟값을 구해보자.
1번 집의 색은 2번, N번 집의 색과 같지 않아야 한다.
N번 집의 색은 N-1번, 1번 집의 색과 같지 않아야 한다.
i(2 ≤ i ≤ N-1)번 집의 색은 i-1, i+1번 집의 색과 같지 않아야 한다.
입력
첫째 줄에 집의 수 N(2 ≤ N ≤ 1,000)이 주어진다. 둘째 줄부터 N개의 줄에는 각 집을 빨강, 초록, 파랑으로 칠하는 비용이 1번 집부터 한 줄에 하나씩 주어진다. 집을 칠하는 비용은 1,000보다 작거나 같은 자연수이다.
출력
첫째 줄에 모든 집을 칠하는 비용의 최솟값을 출력한다.
예제 입력 1
1
2
3
4
| 3
26 40 83
49 60 57
13 89 99
|
예제 출력 1
예제 입력 2
1
2
3
4
| 3
1 100 100
100 1 100
100 100 1
|
예제 출력 2
예제 입력 3
1
2
3
4
| 3
1 100 100
100 100 100
1 100 100
|
예제 출력 3
예제 입력 4
1
2
3
4
5
6
7
| 6
30 19 5
64 77 64
15 19 97
4 71 57
90 86 84
93 32 91
|
예제 출력 4
예제 입력 5
1
2
3
4
5
6
7
8
9
| 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
|
예제 출력 5
내 코드
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
| const [N, ...input] = require("fs")
.readFileSync("/dev/stdin")
.toString()
.trim()
.split("\n");
const n = +N;
const costs = input.map((elem) => elem.split(" ").map(Number));
const MAX = 100001;
let answer = MAX;
let dp = Array.from({ length: n + 1 }, () => [0, 0, 0]);
for (let i = 0; i < 3; i++) {
for (let j = 0; j < 3; j++) {
if (i === j) dp[0][j] = costs[0][j];
else dp[0][j] = MAX;
}
for (let j = 1; j < n; j++) {
dp[j][0] = Math.min(dp[j - 1][1], dp[j - 1][2]) + costs[j][0];
dp[j][1] = Math.min(dp[j - 1][0], dp[j - 1][2]) + costs[j][1];
dp[j][2] = Math.min(dp[j - 1][0], dp[j - 1][1]) + costs[j][2];
}
for (let j = 0; j < 3; j++) {
if (i === j) continue;
answer = Math.min(answer, dp[n - 1][j]);
}
}
console.log(answer);
|
해설
RGB 거리 문제와는 다르게 첫번째 선택한 색상을 마지막에서도 선택하면 안되므로 첫번째 선택한 색상을 기억해두도록 접근하자.
예를 들어 첫번째 색상을 R로 선택했다고 하면 나머지 G,B 색상은 선택하지 못하도록 최대값으로 설정해준다.
이후에 다음 집에서는 R을 제외한 나머지 2가지 색상 중 작은 값을 선택하고 해당 집의 색상 비용을 더해줘서 DP 배열을 완성 시켜나간다.
그리하여 DP 배열이 완성되고 나면 DP 배열의 최종 인덱스에 위치한 배열 중 가장 작은 값을 answer 변수에 저장해둔다.
이렇게 첫번째 집을 R로 설정한 경우, G로 설정한 경우, B로 설정한 경우 총 3번을 진행하고 나서 최종적으로 answer 값을 출력해주면 된다.