[BOJ][Python] 2579번-계단 오르기

[문제 및 출처]

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

2579번: 계단 오르기

[어떻게 풀까?]

이 문제에 대해 이해하기 위해서  간단하게 그림을 그려보았다.

우리가 i번째 계단에 가기 위해서는 저런 식으로 가야한다. 왜냐하면 3계단을 연속으로 오를 수 없기 때문!

이것을 동적 계획법과 연관지어 어떻게 풀어야 할까?

경우의 수를 누적시키면 된다. 코드를 보면 이해가 될 것이다.

[코드]

n = int(input())
array = [0 for i in range(301)]
dp = [0 for i in range(301)]

for i in range(n):
    array[i] = int(input())

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

print(dp[n-1])