Swift - 백준 2156번 포도주 시식

그려보고 공식을 찾아보자, 값을 건너뛰면서 누적시킬 수 있다

 

 

import Foundation

var input = Int(readLine()!)!
var arr: [Int] = .init(repeating: 0, count: 3)
var dp: [Int] = .init(repeating: 0, count: 3)

for _ in 0..<input {
    arr.append(Int(readLine()!)!)
}

for i in 3..<arr.count {
    let tmp = max(dp[i-2]+arr[i], max(dp[i-3]+arr[i-1]+arr[i], dp[i-1]))
    dp.append(tmp)
}

print(dp.max()!)