[백준 2579 파이썬] 계단 오르기

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

# 계단오르기
# 참고 : https://daimhada.tistory.com/181

import sys
input = sys.stdin.readline

n = int(input().rstrip())
stair = [0] * 301
for i in range(n):
    stair[i] = int(input().rstrip())

dp = [0] * 301
dp[0] = stair[0] # 첫번째 계단
dp[1] = stair[0] + stair[1] # 두번째 계단
dp[2] = max(stair[0] + stair[2], stair[1] + stair[2]) # 세번째 계단
# 네번째 계단 : dp[3] = stair[0] + stair[1] + stair[3] 혹은 stair[0] + stair[2] + stair[3]

for i in range(3, n):
    # 한 칸전 계단을 밟고 올라온 경우, 두칸 전 계단을 밟고 올라온 경우
    dp[i] = max(dp[i - 3] + stair[i - 1] + stair[i], dp[i - 2] + stair[i])

print(dp[n - 1]) 

참고
https://daimhada.tistory.com/181