백준 13398 [kotlin]

문제

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

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

만약, -35를 제거한다면, 수열은 10, -4, 3, 1, 5, 6, 12, 21, -1이 되고, 여기서 정답은 10-4+3+1+5+6+12+21인 54가 된다.

입력

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

출력

첫째 줄에 답을 출력한다.

풀이

첫번째 시행 착오는 이차원 배열을 사용하여서 

1~n까지의 배열을 모두 2차원 배열에 저장을 해두었는데 이렇게 하게 된다면, (제외한 값) * (스타트 위치) 가 된다면 

dp 사이즈가 100000 * 100000 이므로 메모리 초과가 발생하였다.

 

두번째 시도는 그래서 메모리를 최대한 줄이기 위해 

[i까지에서의 최대값][삭제 했는가 안했는가] 이렇게 하여서 dp 사이즈를 100000 * 2 로 줄여서 해결하였다.

 

코드

import java.util.*
import kotlin.math.max

fun main()
{
    val n = readLine()!!.toInt()
    val ar = readLine()!!.split(" ").map {it.toInt()}

    var dp = Array(100001,{IntArray(2,{0})})

    var ans = ar[0]
    for(i in 0..n-1){
        dp[i][0] = ar[i]
        dp[i][1] = ar[i]
        if(i == 0) continue
        dp[i][0] = max(dp[i-1][0] + ar[i], ar[i])
        dp[i][1] = max(dp[i-1][0],dp[i-1][1] + ar[i])
        ans = maxOf(ans, dp[i][0], dp[i][1])
    }
    println(ans)
}