[백준] 1149 RGB거리 - Dynamic Programming / Java

• 문제 링크

1149번: RGB거리

 

• 풀이 과정

dp 배열을 2차원 배열로 생성하고, 이는 dp[집][색상] 을 의미한다.

 

이후 각각의 집에 색을 칠하는 비용을 입력받으며,

무조건 첫번째 집에 가장 적은 비용의 색을 칠함에 따라 정답을 구할 수 있는 것이 아니기 때문에

첫번째 집에 3가지 색을 칠한 경우의 수를 모두 구한다.

 

현재 집까지 색을 칠하는 최소 비용은,

이전 집에서 현재 집과 다른 두 색상 중 적은 비용과 칠할 색의 비용을 합한 값이므로

i번째 집에서 빨간색을 칠할 경우 dp[i][0] = Math.min(dp[i - 1][1], dp[i - 1][2]) + red 로 나타낼 수 있다.

 

이렇게 각 시작점에서 최소 비용으로 색을 칠할 수 있는 모든 경우의 수를 구하고,

n번째 집에서의 최소 비용 Math.min(dp[n][0], Math.min(dp[n][1], dp[n][2])) 을 정답으로써 출력한다.

 

• 풀이 코드

import java.io.BufferedReader;
import java.io.BufferedWriter;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.OutputStreamWriter;
import java.util.StringTokenizer;

public class Main {

    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));
        StringTokenizer st;

        int n = Integer.parseInt(br.readLine());
        int[][] dp = new int[n + 1][3];

        for (int i = 1; i <= n; i++) {
            st = new StringTokenizer(br.readLine());
            int red = Integer.parseInt(st.nextToken());
            int green = Integer.parseInt(st.nextToken());
            int blue = Integer.parseInt(st.nextToken());

            dp[i][0] = Math.min(dp[i - 1][1], dp[i - 1][2]) + red;
            dp[i][1] = Math.min(dp[i - 1][0], dp[i - 1][2]) + green;
            dp[i][2] = Math.min(dp[i - 1][0], dp[i - 1][1]) + blue;
        }

        bw.write(Math.min(dp[n][0], Math.min(dp[n][1], dp[n][2])) + "\n");
        bw.flush();
    }

}