[Java/자바][백준 1149: RGB 거리]

https://www.acmicpc.net/problem/1149

1149번: RGB거리

 

Problem

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

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

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

 

Input

첫째 줄에 집의 수 N(2 ≤ N ≤ 1,000)이 주어진다. 둘째 줄부터 N개의 줄에는 각 집을 빨강, 초록, 파랑으로 칠하는 비용이 1번 집부터 한 줄에 하나씩 주어진다. 집을 칠하는 비용은 1,000보다 작거나 같은 자연수이다.

 

Output

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

 

Insight

1. DP(동적 계획법)을 통해서 문제를 해결할 수 있다. 

2. 연속하지 않으면서 가장 적은 비용으로 N개의 집을 칠해야 한다. 

3. memoization 배열에 i번째 집을 R,G,B로 색칠 할 때의 가장 최소값을 저장한다. 

• d[i][0]: 빨강으로 칠할 때의 최소비용  = 이웃한 집(i-1)이 초록 혹은 파랑으로 칠해져있어야 함 =  min(d[i-1][1], d[i-1][2]) + arr[i] 
  
  
• d[i][1]: 초록으로 칠할 때의 최소비용  = 이웃한 집이 빨강 혹은 파랑으로 칠해져있어야 함 =  min(d[i-1][0], d[i-1][2]) + arr[i] 
  
  
• d[i][2]: 파랑으로 칠할 때의 최소비용  = 이웃한 집이 빨강 혹은 초록으로 칠해져있어야 함 =  min(d[i-1][0], d[i-1][1]) + arr[i] 

 

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

public class Main {
	
    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[][] d = new int[N+1][3]; 
        int[][] arr = new int[N+1][3];
        
        for(int i=1; i<=N; i++) {
        	StringTokenizer st= new StringTokenizer(br.readLine()," ");
        	for(int j=0; j<3; j++) {
        		arr[i][j] = Integer.parseInt(st.nextToken());
        	}
        }
        
        for(int i=1; i<=N; i++) {
        	if(i==1) {
        		d[i][0] = arr[i][0]; 
        		d[i][1] = arr[i][1];
        		d[i][2] = arr[i][2];	
        	}
        	else {
        		d[i][0] = Math.min(d[i-1][1], d[i-1][2])+arr[i][0];
        		d[i][1] = Math.min(d[i-1][0], d[i-1][2])+arr[i][1];
        		d[i][2] = Math.min(d[i-1][0], d[i-1][1])+arr[i][2];
        	}
        }
        
        int res = Math.min(d[N][0], d[N][1]);
        res = Math.min(res, d[N][2]);
           
        bw.write(String.valueOf(res));
        		
        
        
        bw.flush();
        bw.close();
    }

}