[백준] 2579번 : 계단 오르기 - Swift

문제

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

2579번: 계단 오르기

코드

var num = Int(readLine()!)
var arr = [Int]()
var dp = [Int](repeating: 0, count: num!)

for i in 0 ..< num!
{
    let input = Int(readLine()!)
    arr.append(input!)
    if i == 0{dp[0] = arr[0]}
    else if i == 1 {dp[1] = max(dp[0]+arr[1],arr[1])}
    else if i == 2 {dp[2] = max(dp[0]+arr[2],arr[1]+arr[2])}
    else {dp[i] = max(dp[i-3]+arr[i-1]+arr[i], dp[i-2]+arr[i])}
}


print(dp[num!-1])