[Python/Kotlin] 백준 1149번 : RGB 거리

백준 문제 주소 https://www.acmicpc.net/problem/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보다 작거나 같은 자연수이다.

3
26 40 83
49 60 57
13 89 99

출력

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

96

---

문제에서 집이 한 줄로 지어져있기 때문에 옆 집들의 색과 안겹치면 된다. 이 문제에서 반복되는 규칙을 찾는다면, 전의 값과 현재의 값을 더해서 최종 최솟값을 찾는 것이다. 하지만 순간순간의 최솟값을 선택했다고 하더라도 최종적으로는 최솟 값이 아닐 수 있다. 따라서 저장을 하면서 계산을 해야한다.

그래서 빨강, 초록, 파랑으로 다 시작을 하며 저장한다고 생각하면 점화식은 아래와 같다.

dp[i][0] = min(dp[i-1][1], dp[i-1][2]) + arr[i][0]
dp[i][1] = min(dp[i-1][0], dp[i-1][2]) + arr[i][1]
dp[i][2] = min(dp[i-1][0], dp[i-1][1]) + arr[i][2]

Python 풀이

n = int(input())
arr = [[0 for _ in range(3)] for _ in range(n)]
dp = [[0 for _ in range(3)] for _ in range(n)]

for i in range(n):
    arr[i][0], arr[i][1], arr[i][2] = map(int, input().split())

dp[0][0] = arr[0][0]
dp[0][1] = arr[0][1]
dp[0][2] = arr[0][2]

for i in range(1, n):
    dp[i][0] = min(dp[i-1][1], dp[i-1][2]) + arr[i][0]
    dp[i][1] = min(dp[i-1][0], dp[i-1][2]) + arr[i][1]
    dp[i][2] = min(dp[i-1][0], dp[i-1][1]) + arr[i][2]
    
print(min(dp[n-1][0], dp[n-1][1], dp[n-1][2]))

Kotlin 풀이

import java.io.*
import java.util.*

fun main(args: Array<String>)
{
    val br = BufferedReader(InputStreamReader(System.`in`))
    val n = br.readLine().toInt()
    val arr = Array(n){Array(3){0} }
    val dp = Array(n){Array(3){0} }

    for (i in 0 until n) {
        val st = StringTokenizer(br.readLine())
        arr[i][0] = st.nextToken().toInt()
        arr[i][1] = st.nextToken().toInt()
        arr[i][2] = st.nextToken().toInt()
    }

    dp[0][0] = arr[0][0]
    dp[0][1] = arr[0][1]
    dp[0][2] = arr[0][2]

    for (i in 1 until n) {
        dp[i][0] = Math.min(dp[i-1][1], dp[i-1][2]) + arr[i][0]
        dp[i][1] = Math.min(dp[i-1][0], dp[i-1][2]) + arr[i][1]
        dp[i][2] = Math.min(dp[i-1][0], dp[i-1][1]) + arr[i][2]
    }

    println(Math.min(dp[n-1][0], Math.min(dp[n-1][1], dp[n-1][2])))
    
}