[백준] 2579번:계단 오르기 (python 파이썬)

문제

www.acmicpc.net/problem/2579

2579번: 계단 오르기

 

풀이 및 소스코드

편의상 1번째 계단에 대한 점수는 stiars[1]에 있는 것처럼 설명했지만,

코드 짤때는 stairs, dp 모두 0~n-1 인덱스로 초기화했다!

import sys

input = sys.stdin.readline

n = int(input())
stairs = []
dp = []
for i in range(n):
    stairs.append(int(input()))

if n==1:
    print(stairs[0])
    exit()
elif n == 2:
    print(max(stairs[0]+stairs[1], stairs[1]))
    exit()

dp.append(stairs[0])
dp.append(max(dp[0]+stairs[1], stairs[1]))
dp.append(max(dp[0]+stairs[2], stairs[1]+stairs[2]))

for i in range(3, n):
    dp.append(max(dp[i-2]+stairs[i], dp[i-3]+stairs[i-1]+stairs[i]))

print(dp[-1])