[백준][코틀린] 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보다 작거나 같은 자연수이다.

출력

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

예제 입력	예제 출력
3 26 40 83 49 60 57 13 89 99	96
3 1 100 100 100 1 100 100 100 1	3
3 1 100 100 100 100 100 1 100 100	102

문제 풀이

본 문제에서 각 집을 칠해서 가는데 최소 비용을 구해가야한다. 따라서 첫 집부터 칠하는 비용을 누적해서 가야되는데 최소 비용을 구해야 된다는 것이다. 그럼 케이스를 나눠서 점화식을 구해야되는데, 본 문제는 3가지 경우가 있다.

• index 번째 집을 빨간색(0)으로 -> min (index-1번째 집이 파란색(1)인 경우, index-1번째 집이 초록색(2)인 경우)
• index 번째 집을 파란색(1)으로 -> min (index-1번째 집이 빨간색(0)인 경우, index-1번째 집이 초록색(2)인 경우)
• index 번째 집을 초록색(2)으로 -> min (index-1번째 집이 빨간색(0)인 경우, index-1번째 집이 파란색(1)인 경우)

이렇게 마지막 집까지 확인을 확인한 후, 젤 작은 값을 출력하면 된다.

</>̆̈ 코드

import java.util.*

fun main() {
    val br = System.`in`.bufferedReader()
    val n = br.readLine().toInt()
    val dp = Array(n) {StringTokenizer(br.readLine()).run { IntArray(3) {nextToken().toInt()} }}

    for (index in 1 until n) {
        dp[index][0] += minOf(dp[index-1][1], dp[index-1][2])
        dp[index][1] += minOf(dp[index-1][0], dp[index-1][2])
        dp[index][2] += minOf(dp[index-1][0], dp[index-1][1])
    }

    print(minOf(dp[n-1][0], dp[n-1][1], dp[n-1][2]))

}

링크

1149번: RGB거리