[Swift] 백준 정수 삼각형

문제: https://www.acmicpc.net/problem/1932

1932번: 정수 삼각형

 

코드

import Foundation

let N = Int(readLine()!)!
var dp = [[0, 0]]

for _ in 0..<N {
    dp.append(readLine()!.split(separator: " ").map { Int($0)! })
}

for i in 1..<dp.count {
    for j in 0..<dp[i].count {
        if j == 0 { // 맨 왼쪽거는
            dp[i][j] = dp[i][j] + dp[i - 1][j]
        } else if j == dp[i].count - 1 { // 맨 오른쪽 거
            dp[i][j] = dp[i][j] + dp[i - 1][j - 1]
        } else {
            dp[i][j] = max(dp[i][j] + dp[i - 1][j], dp[i][j] + dp[i - 1][j - 1])
        }
    }
}

//print(dp)

var result = dp[N].max()!

print(result)