[伯俊]2096号:下りて解答する


質問リンク
https://www.acmicpc.net/problem/2096
解き方
この問題はDP問題であり、点火方式は以下の通りである.
maxDP1[i] = arr[i][0] + max(maxDP1[i-1], maxDP2[i-1])
maxDP2[i] = arr[i][1] + max(maxDP1[i-1], maxDP2[i-1], maxDP3[i-1])
maxDP3[i] = arr[i][2] + max(maxDP2[i-1], maxDP3[i-1])
minDPも同様に行う.
しかし、問題のメモリ条件が非常に厳しいため、従来の方法でDPを使用するとメモリが過剰になるという問題が発生します.
そのため、スライドウィンドウ技術を使用する必要があります.
スライドウィンドウとは?
dpでコメント作成を行う場合、未使用の値を配列に格納するのではなく、配列の更新を続行します.
そのため、点火式を以下のように再整理することができます.
maxDP = [arr[0] + max(maxDP[0], maxDP[1]), arr[1] + max(maxDP), arr[2] + max(maxDP[1], maxDP[2])]
minDPも同様に行われる.
完全なコード
from sys import stdin

N = int(input())
# 맨 처음 세개의 숫자를 입력받아 DP의 초기 값을 설정한다.
arr = list(map(int, stdin.readline().split()))
maxDP = arr
minDP = arr
for _ in range(N - 1):
    arr = list(map(int, stdin.readline().split()))
    # 세가지 값을 입력받을 때마다, DP에 새롭게 갱신한다.
    maxDP = [arr[0] + max(maxDP[0], maxDP[1]), arr[1] + max(maxDP), arr[2] + max(maxDP[1], maxDP[2])]
    minDP = [arr[0] + min(minDP[0], minDP[1]), arr[1] + min(minDP), arr[2] + min(minDP[1], minDP[2])]

print(max(maxDP), min(minDP))