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

문제

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

2579번: 계단 오르기

 

코드

n=int(input())
array=[0]
for _ in range(n):
    array.append(int(input()))
if n==1:
    print(array[1])
else:

    dp=[0]*(n+1)
    dp[1]=array[1]
    dp[2]=array[1]+array[2]

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

    print(dp[n])