[BOJ] 2096. 내려가기

2096. 내려가기

1. 문제

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

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

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

2. 입력

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

3. 출력

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

4. 풀이

C++

#include <iostream>
#include <algorithm>
using namespace std;
#define ioboost ios::sync_with_stdio(false); cout.tie(NULL); cin.tie(NULL);

int N;

// dp 배열은 메모리 제한 때문에 1차원 배열로 해서 계속 갱신
int arr[100001][3], dp_max[3], dp_min[3];

void Input()
{
    cin >> N;
    for (int i = 1; i <= N; ++i)
        cin >> arr[i][0] >> arr[i][1] >> arr[i][2];
}

void Solve()
{
    dp_max[0] = arr[1][0];
    dp_max[1] = arr[1][1];
    dp_max[2] = arr[1][2];
    
    dp_min[0] = arr[1][0];
    dp_min[1] = arr[1][1];
    dp_min[2] = arr[1][2];
    
    int tmp0, tmp1, tmp2;
    for (int i = 2; i <= N; ++i)
    {
        tmp0 = dp_max[0];
        tmp1 = dp_max[1];
        tmp2 = dp_max[2];
        
        dp_max[0] = arr[i][0] + max(tmp0, tmp1);
        dp_max[1] = arr[i][1] + max(max(tmp0, tmp1), tmp2);
        dp_max[2] = arr[i][2] + max(tmp1, tmp2);
        
        tmp0 = dp_min[0];
        tmp1 = dp_min[1];
        tmp2 = dp_min[2];
        
        dp_min[0] = arr[i][0] + min(tmp0, tmp1);
        dp_min[1] = arr[i][1] + min(min(tmp0, tmp1), tmp2);
        dp_min[2] = arr[i][2] + min(tmp1, tmp2);
    }
    
    cout << max(max(dp_max[0], dp_max[1]), dp_max[2]) << " " << min(min(dp_min[0], dp_min[1]), dp_min[2]);
}


int main()
{
    ioboost;
    Input();
    Solve();
}