백준 1149 RGB 거리 Java DP 풀이

 

 

문제

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

출력

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

---

문제 해설

이 문제는 그냥 재귀로 풀면 안되나..? 싶었는데 DP 연습문제라고 받은 거라서 열심히 고민을 했다.

사실 그 고민의 끝에 답이 있지는 않아서 도움을 받아 풀이하게 되었다.

각각의 집이 각각의 모든 색을 칠하는 최적 값을 찾아가며 풀이하게 되는 문제였다.

 

이해하기 어려웠는데... 코드는 엄청나게 짧아서 화가 났다.

이렇게 짧고 쉽고 어려운 문제는 처음이라서 많이 슬펐다.

 

제출 코드

 

package Dynamic_Programing;

import java.io.BufferedReader;
import java.io.File;
import java.io.FileReader;
import java.io.InputStreamReader;
import java.util.StringTokenizer;

public class BOJ_1149_RGB거리 {
    static int[][] houseArr;
    static int N;
    public static void main(String[] args) throws Exception {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        N = Integer.parseInt(br.readLine());

        houseArr = new int[3][N];
        StringTokenizer st= new StringTokenizer(br.readLine());
        houseArr[0][0] = Integer.parseInt(st.nextToken());
        houseArr[1][0] = Integer.parseInt(st.nextToken());
        houseArr[2][0] = Integer.parseInt(st.nextToken());
        for (int i = 1; i < N; i++) {
            st= new StringTokenizer(br.readLine());
            int r = Integer.parseInt(st.nextToken());
            int g = Integer.parseInt(st.nextToken());
            int b = Integer.parseInt(st.nextToken());

            houseArr[0][i] = Math.min(houseArr[1][i-1], houseArr[2][i-1])+r;// 앞집이 파란색, 초록색 집일 경우 최소치에 빨간색 칠함.
            houseArr[1][i] = Math.min(houseArr[0][i-1], houseArr[2][i-1])+g;
            houseArr[2][i] = Math.min(houseArr[0][i-1], houseArr[1][i-1])+b;
        }
        System.out.println(Math.min(houseArr[0][N-1], Math.min(houseArr[1][N-1], houseArr[2][N-1])));

    }
}