백준 17404번 - RGB거리 2

문제

백준 17404번 - RGB거리 2

아이디어

• 1번 집부터 빨강 또는 초록 또는 파랑으로 칠하기 시작했을 때 각각을 비교해야 한다.
• dp[i][j]를 i번 집까지 탐색했을 때 j 색깔(빨강0, 초록1, 파랑2)의 최소비용이라 가정한다.
• 딱 세번 반복하여 각 색깔로 시작하는 경우 해당값을, 나머지는 최댓값을 넣어주어 내려가면서 비교를 한다.
• 같은 열만 신경쓰지 않으면 1번과 N번은 다른 색이어야 한다라는 조건이 맞게 된다.
• 마지막 집까지 결과를 구했다면 1번과 N번이 다른 색인 경우에만 최소비용을 비교해 갱신해준다.

예상 시간 복잡도

• 예상 시간 복잡도 : O(N)

코드 구현

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

public class BJ_17404 {
    public static void main(String[] args) throws IOException {

        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));

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

        int[][] cost = new int[n][3];
        int[][] dp = 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());
            }
        }

        int min = Integer.MAX_VALUE;

        for (int i = 0; i < 3; i++) { //1번 집의 색깔

            //1번 집 최소 비용 설정
            for (int j = 0; j < 3; j++) {
                if (i == j) {
                    dp[0][j] = cost[0][j];
                } else {
                    dp[0][j] = 1000 * 1000;
                }
            }

            //인접한 집끼리 서로 다른 색깔로 칠해야 한다.
            for (int j = 1; j < n; j++) {
                dp[j][0] = Math.min(dp[j - 1][1], dp[j - 1][2]) + cost[j][0];
                dp[j][1] = Math.min(dp[j - 1][0], dp[j - 1][2]) + cost[j][1];
                dp[j][2] = Math.min(dp[j - 1][0], dp[j - 1][1]) + cost[j][2];
            }

            //마지막 집에서 1번 집과 다른 색깔일 때 최소비용 비교 후 갱신
            for (int j = 0; j < 3; j++) {
                if (i != j) {
                    min = Math.min(min, dp[n - 1][j]);
                }
            }
        }

        System.out.println(min);
    }
}