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

---

🗝 풀이 (언어 : Java)

DP로 푸는 문제이다. 위, 아래 인접한 집과 색이 겹치지만 않으면 된다. 둘째 줄부터 돌면서 해당 위치 색깔과 다른 색깔의 윗줄의 두개의 비용 중에 최소 비용을 해당 위치의 비용과 더하면서 끝까지 내려간다. 마지막 줄의 3개의 비용 중에 최소의 비용이 정답이다.

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

public class Main {

    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        int n = Integer.parseInt(br.readLine());
        int[][] matrix = new int[n][3];
        for (int i = 0; i < n; i++) {
            StringTokenizer st2 = new StringTokenizer(br.readLine(), " ");
            for (int j = 0; j < 3; j++) {
                matrix[i][j] = Integer.parseInt(st2.nextToken());
            }
        }
        br.close();
        System.out.println(countMoney(matrix));
    }

    public static int countMoney(int[][] matrix) {
        for (int i = 1; i < matrix.length; i++) {
            for (int j = 0; j < 3; j++) {
                if (j == 0) {
                    matrix[i][j] += Math.min(matrix[i - 1][1], matrix[i - 1][2]);
                }
                if (j == 1) {
                    matrix[i][j] += Math.min(matrix[i - 1][0], matrix[i - 1][2]);
                }
                if (j == 2) {
                    matrix[i][j] += Math.min(matrix[i - 1][0], matrix[i - 1][1]);
                }
            }
        }
        return Math.min(Math.min(matrix[matrix.length - 1][0], matrix[matrix.length - 1][1]), matrix[matrix.length - 1][2]);
    }

}