2096번: 내려가기 (BOJ C++)

2096번:  내려가기

사용 언어: C++

 

문제

 

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

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

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

입력

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

출력

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

메모리 제한

• Java 8: 256 MB
• Java 8 (OpenJDK): 256 MB
• Java 11: 256 MB
• Kotlin (JVM): 256 MB

 

풀이

1149번: RGB거리와 달리 배열을 미리 다 만들어 놓으면 dp[100000][3]와 같은 배열이 3개나 나와서 메모리 제한에 걸려서 배열을 [0]과 [1]만 사용해서 계산을 해가면서 풀었다.

#include <bits/stdc++.h>
using namespace std;

int main(void)
{
    ios::sync_with_stdio(0);
    cin.tie(0);
    
    int n;
    cin >> n; 
	int arr[3];
	int dp_max[2][3], dp_min[2][3];
	int i=0;
	for(int j=0; j<3; j++)
			cin >> arr[j];
	dp_max[0][0] = dp_min[0][0] = arr[0];
	dp_max[0][1] = dp_min[0][1] = arr[1];
	dp_max[0][2] = dp_min[0][2] = arr[2];
	
	//
	while(i < n-1)
	{
		for(int j=0; j<3; j++)
			cin >> arr[j];
		dp_max[1][0] = arr[0] + max(dp_max[0][0], dp_max[0][1]);
		dp_max[1][1] = arr[1] + max({dp_max[0][0], dp_max[0][1], dp_max[0][2]});
        dp_max[1][2] = arr[2] + max(dp_max[0][1], dp_max[0][2]);
		dp_min[1][0] = arr[0] + min(dp_min[0][0], dp_min[0][1]);
		dp_min[1][1] = arr[1] + min({dp_min[0][0], dp_min[0][1], dp_min[0][2]});
        dp_min[1][2] = arr[2] + min(dp_min[0][1], dp_min[0][2]);
		dp_max[0][0] = dp_max[1][0];
		dp_max[0][1] = dp_max[1][1];
		dp_max[0][2] = dp_max[1][2];
		dp_min[0][0] = dp_min[1][0];
		dp_min[0][1] = dp_min[1][1];
		dp_min[0][2] = dp_min[1][2];
		i++;
	}
	int max_sum, min_sum = 0;
	
	//입력이 하나일 경우
	if(n == 1)
	{
		max_sum = max({dp_max[0][0], dp_max[0][1], dp_max[0][2]});
		min_sum = min({dp_min[0][0], dp_min[0][1], dp_min[0][2]});		
	}
	//입력이 여러 개일 경우
	else
	{
		max_sum = max({dp_max[1][0], dp_max[1][1], dp_max[1][2]});
		min_sum = min({dp_min[1][0], dp_min[1][1], dp_min[1][2]});		
	}

	cout << max_sum << " " << min_sum << '\n';	

	return 0; 
}