2022.03.24 89일차

백준 1149번 (RGB거리)

문제

RGB거리에는 집이 N개 있다. 거리는 선분으로 나타낼 수 있고, 1번 집부터 N번 집이 순서대로 있다.

집은 빨강, 초록, 파랑 중 하나의 색으로 칠해야 한다. 각각의 집을 빨강, 초록, 파랑으로 칠하는 비용이 주어졌을 때, 아래 규칙을 만족하면서 모든 집을 칠하는 비용의 최솟값을 구해보자.

1번 집의 색은 2번 집의 색과 같지 않아야 한다.
N번 집의 색은 N-1번 집의 색과 같지 않아야 한다.
i(2 ≤ i ≤ N-1)번 집의 색은 i-1번, i+1번 집의 색과 같지 않아야 한다.

나의 풀이

1. 조건1 - 색이 이어서 같은 색이면 안된다. 조건2 - 비용은 최소
2. 경우에 따라 이전 값이 변경되어야 할 수 있으므로 전체의 경우 구하기
3. n번째 값에서 인덱스가 겹치지 않는 두 수 중 최소값 구하기, 구한 값과 n번째 값 더하기
4. 점화식
   ans[n][0] = Math.min(ans[n-1][1], ams[n-1][2]) + ans[n][0]
   ans[n][1] = Math.min(ans[n-1][0], ams[n-1][2]) + ans[n][1]
   ans[n][2] = Math.min(ans[n-1][0], ams[n-1][1]) + ans[n][2]
5. 전체의 경우 중 최소값 출력

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

public class Main {
    static int[][] dp;
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));

        int n = Integer.parseInt(br.readLine());

        dp = new int[n + 1][3];
        int[][] nArr = new int[n + 1][3];

        for (int i = 1; i <= n; i++) {
            StringTokenizer st = new StringTokenizer(br.readLine());
            nArr[i][0] = Integer.parseInt(st.nextToken());
            nArr[i][1] = Integer.parseInt(st.nextToken());
            nArr[i][2] = Integer.parseInt(st.nextToken());
        }

        for (int i = 1; i <= n; i++) {
            dp[i][0] = Math.min(dp[i - 1][1], dp[i - 1][2]) + nArr[i][0];
            dp[i][1] = Math.min(dp[i - 1][0], dp[i - 1][2]) + nArr[i][1];
            dp[i][2] = Math.min(dp[i - 1][0], dp[i - 1][1]) + nArr[i][2];
        }

        int min = Math.min(Math.min(dp[n][0], dp[n][1]), dp[n][2]);

        System.out.println(min);
    }
}

생각하기

• DP 문제 풀이 연습