[백준 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가지 시작점 (빨강, 초록, 파랑) 으로 출발한다.
  • 칠한 색깔과 겹치지 않은 색깔의 전체 비용 중 더 적은 비용과 현재 집을 칠할 색깔 비용을 더해준다.
• 그림을 통해 확인해보자.

 

Code

public static void main(String[] args) throws IOException {
    BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
    BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));
    int N = Integer.parseInt(br.readLine());
    int[][] cost = new int[N][3];
    int[][] sum = new int[N][3];
 
    for (int i = 0; i < N; i++) {
        StringTokenizer st = new StringTokenizer(br.readLine());
 
        for (int j = 0; j < 3; j++) {
            cost[i][j] = Integer.parseInt(st.nextToken());
 
            if (i == 0)
                sum[i][j] = cost[i][j];
        }
    }
 
    for (int i = 1; i < N; i++) {
        sum[i][0] = Math.min(sum[i - 1][1], sum[i - 1][2]) + cost[i][0];
        sum[i][1] = Math.min(sum[i - 1][0], sum[i - 1][2]) + cost[i][1];
        sum[i][2] = Math.min(sum[i - 1][0], sum[i - 1][1]) + cost[i][2];
    }
 
    int min = Math.min(Math.min(sum[N - 1][0], sum[N - 1][1]), sum[N - 1][2]);
 
    bw.write(min + " ");
    bw.flush();
    bw.close();
    br.close();
}

 

결과