[백준][Python] 2597번 계단 오르기 - DP

https://www.acmicpc.net/problem/2579

2579번: 계단 오르기

 

 

Python

import sys
input = sys.stdin.readline

n = int(input().rstrip())
arr = [0]+[int(input().rstrip()) for _ in range(n)]

dp = [[arr[i], arr[i]] for i in range(n+1)]

for i in range(2, n+1):
    dp[i][0] += dp[i-1][1]  # 직전 계단은 건너 뛰어서 온 계단만 가능
    dp[i][1] += max(dp[i-2]) # 두 칸 뒤의 계단은 어디서 오든 상관 없음

print(max(dp[-1]))