[Baekjoon] 2096번 내려가기

문제

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

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

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

입력

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

출력

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

예제

Example 1:

Input: 
3
1 2 3
4 5 6
4 9 0
Output: 
18 6

조건

시간 제한 : 1초

메모리 제한 : 4 MB

풀이과정

풀이

문제를 보자마자 전형적인 다이나믹 프로그래밍 문제인데..?! 이게 왜 골드4짜리지?하고 그냥 냈다가 메모리 초과를 받았다. 메모리 제한이 4MB…?!이다.

이 문제는 그냥 DP를 크게 쌓을 필요없이 계속 갱신해가면 되는 문제이다. 느낌은 RGB거리문제와 비슷한 것같다. 이 문제도 이렇게 메모리를 줄일 수 있지 않을까 생각된다.

import sys
input = sys.stdin.readline

n = int(input())

a, b, c = map(int,input().split())

ans1 = [a,b,c]
ans2 = [a,b,c]

for _ in range(n-1):
    x, y, z = map(int, input().split())

    # 최대 계산
    sub1 = [max(ans1[0], ans1[1]) + x, max(ans1[0], ans1[1], ans1[2]) + y, max(ans1[1], ans1[2]) + z]
    ans1 = sub1
    # 최소 계산
    sub2 = [min(ans2[0], ans2[1]) + x, min(ans2[0], ans2[1], ans2[2]) + y, min(ans2[1], ans2[2]) + z]
    ans2 = sub2

print(max(ans1), min(ans2))