1912번: 연속합 - Kotlin

1912번: 연속합

문제

n개의 정수로 이루어진 임의의 수열이 주어진다. 우리는 이 중 연속된 몇 개의 수를 선택해서 구할 수 있는 합 중 가장 큰 합을 구하려고 한다. 단, 수는 한 개 이상 선택해야 한다.

예를 들어서 10, -4, 3, 1, 5, 6, -35, 12, 21, -1 이라는 수열이 주어졌다고 하자. 여기서 정답은 12+21인 33이 정답이 된다.

입력

첫째 줄에 정수 n(1 ≤ n ≤ 100,000)이 주어지고 둘째 줄에는 n개의 정수로 이루어진 수열이 주어진다. 수는 -1,000보다 크거나 같고, 1,000보다 작거나 같은 정수이다.

출력

첫째 줄에 답을 출력한다.

 

 

나의 풀이

 이 문제는 DP를 사용해서 해결해야 하는 문제이다. DP 문제는 언제나와 같이 점화식을 세우는 것이 중요하다. 이 문제는 부분합을 구하는 방식에서 최대값을 구하면 되는 방식이다. 그래서 생각보다 간단하게 해결할 수 있다.

import kotlin.math.max

fun main() = with(System.`in`.bufferedReader()) {
    val n = readLine().toInt()
    val arr = IntArray(n + 1)
    val temp = readLine().split(" ").map { it.toInt() }.toIntArray()

    for (i in 1..n) {
        arr[i] = temp[i - 1]
    }

    val dp = IntArray(n + 1)
    var ans = arr[1]
    dp[1] = arr[1]
    
    for (i in 2 .. n) {
        dp[i] = max(arr[i] + dp[i - 1], arr[i])
        ans = max(ans, dp[i])
    }
    println(ans)
}