본문 바로가기

알고리즘

1149 RGB거리

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

출력

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

풀이

어렵게 생각하지 않고 입력값들 중에서 지그재그로 내려오며 맨 마지막줄에서의 최솟값을 리턴한다.
예를 들면 입력이 다음과 같을 때
26 40 83
49 60 57
13 89 99
첫 줄 26 40 83은 첫줄이므로 위에 값이 없으니 그냥 넘어간다.
두번째 줄부터 차례대로 49를 가지게 되면 자신의 위에 있지 않는 값 즉, 위치(인덱스)가 다른 값들 중 최솟값을 선택하면 된다.
두번째 줄까지 실행하면 결과는 다음과 같다.
26 40 83
89 86 83
13 89 99
위와 같은 방법으로 마지막 줄까지 하면 된다.

코드

import java.io.BufferedReader;  
import java.io.IOException;  
import java.io.InputStreamReader;  
import java.util.Arrays;  
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\[\]\[\] info = new int\[n\]\[3\];  

        for(int i = 0; i < n; i++){  
            StringTokenizer st = new StringTokenizer(br.readLine());  
            for(int j = 0; j < 3; j++){  
                info\[i\]\[j\] = Integer.parseInt(st.nextToken());  
            }  
        }  



        for(int i = 1; i < n; i++){  
            for(int j = 0; j < 3; j++){  
                info\[i\]\[j\] = Math.min(info\[i - 1\]\[(j + 1) % 3\], info\[i-1\]\[(j + 2) % 3\]) + info\[i\]\[j\];  
            }  
        }  

        Arrays.sort(info\[n-1\]);  
        System.out.println(info\[n-1\]\[0\]);  
    }  

}  

'알고리즘' 카테고리의 다른 글

1904 01타일  (0) 2022.03.25
18353 병사 배치하기  (0) 2022.03.25
11057 오르막 수  (0) 2022.03.24
11053 가장 긴 증가하는 부분 수열  (0) 2022.03.24
1697 숨바꼭질  (0) 2022.03.24