[백준] (Swift) 14501번 - 퇴사 (DP로 풀기)

문제링크

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

14501번: 퇴사

 

내가 푼 코드

var n = Int(readLine()!)!
var tp: [[Int]] = []
var dp = Array(repeating: 0, count: 100)

for i in 0..<n {
    tp.append(readLine()!.split(separator: " ").map { Int(String($0))! })
}

for i in 0..<n {
    if dp[i] > dp[i+1] {
        dp[i+1] = dp[i]
    }

    if dp[i+tp[i][0]] < dp[i] + tp[i][1] {
        dp[i+tp[i][0]] = dp[i] + tp[i][1]
    }
}

print(dp[n])