205. 계단 오르기

백준

1. Python

import sys 
input = sys.stdin.readline 
n = int(input())
stair = [0]
for _ in range(n):
    stair.append(int(input()))

if n == 1:
    print(stair[1])
else:
    dp = [0] * (n + 1)
    dp[1] = stair[1]
    dp[2] = stair[1] + stair[2] 

    for i in range(3, n + 1):
        dp[i] = max(dp[i - 3] + stair[i - 1] + stair[i], dp[i - 2] + stair[i])  

    print(dp[n])

• 참고로 입력값 0에 대해서 처리를 하지 않으면 indexed error가 발생한다.

2. C++