[BOJ] 백준 2096번 내려가기 - Java

2096번: 내려가기

문제

N줄에 0 이상 9 이하의 숫자가 세 개씩 적혀 있다. 내려가기 게임을 하고 있는데, 이 게임은 첫 줄에서 시작해서 마지막 줄에서 끝나게 되는 놀이이다.

먼저 처음에 적혀 있는 세 개의 숫자 중에서 하나를 골라서 시작하게 된다. 그리고 다음 줄로 내려가는데, 다음 줄로 내려갈 때에는 다음과 같은 제약 조건이 있다. 바로 아래의 수로 넘어가거나, 아니면 바로 아래의 수와 붙어 있는 수로만 이동할 수 있다는 것이다. 이 제약 조건을 그림으로 나타내어 보면 다음과 같다.

별표는 현재 위치이고, 그 아랫 줄의 파란 동그라미는 원룡이가 다음 줄로 내려갈 수 있는 위치이며, 빨간 가위표는 원룡이가 내려갈 수 없는 위치가 된다. 숫자표가 주어져 있을 때, 얻을 수 있는 최대 점수, 최소 점수를 구하는 프로그램을 작성하시오. 점수는 원룡이가 위치한 곳의 수의 합이다.

입력

첫째 줄에 N(1 ≤ N ≤ 100,000)이 주어진다. 다음 N개의 줄에는 숫자가 세 개씩 주어진다. 숫자는 0, 1, 2, 3, 4, 5, 6, 7, 8, 9 중의 하나가 된다.

출력

첫째 줄에 얻을 수 있는 최대 점수와 최소 점수를 띄어서 출력한다.

 

풀이

이 문제는 DP문제로 시간이 1초임을 유의해서 풀어야 한다.

단순한 아이디어로 풀수 있는 문제이다.

 

왼쪽 끝과 오른쪽 끝은 내려갈 수 있는 곳이 2곳으로 한정되어 있고, 중간은 세 곳 모두 갈 수 있으므로 연산을 최대와 최소만 찾아서 구해주면 답은 나온다.

 

괜히 이 방법이 아닐까봐 다른 방법을 생각해보다가 시간이 오래 걸렸다.

 

 

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

public class Main {
    static int N;
    public static void main(String[] args) throws IOException {
    
        BufferedReader bf = new BufferedReader(new InputStreamReader(System.in));
        N = Integer.parseInt(bf.readLine());
        StringTokenizer st = new StringTokenizer(bf.readLine());
        int[] input = new int[3];
        int[][] dp = new int[N][6];
        
        for (int i = 0; i < 3; i++)
            input[i] = Integer.parseInt(st.nextToken());
       
       for (int i = 0; i < 3; i++) 
            dp[0][3 + i] = dp[0][i] = input[i];
        

        for (int k = 1; k < N; k++) {
        
            st = new StringTokenizer(bf.readLine());
            
            for (int j = 0; j < 3; j++)
                input[j] = Integer.parseInt(st.nextToken());
                
            int i = k-1;
            
            dp[k][0] = Integer.min(dp[i][0], dp[i][1]) + input[0];
            dp[k][1] = Integer.min(dp[i][0], Integer.min(dp[i][1], dp[i][2])) + input[1];
            dp[k][2] = Integer.min(dp[i][1], dp[i][2]) + input[2];

            dp[k][3] = Integer.max(dp[i][3], dp[i][4]) + input[0];
            dp[k][4] = Integer.max(dp[i][3], Integer.max(dp[i][4], dp[i][5])) + input[1];
            dp[k][5] = Integer.max(dp[i][4], dp[i][5]) + input[2];
        }

        Arrays.sort(dp[N-1]);
        System.out.printf("%d %d",dp[N-1][5],dp[N-1][0]);

    }
}