[Python/Kotlin] 백준 1912번 : 연속합

백준 문제 주소 https://www.acmicpc.net/problem/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보다 작거나 같은 정수이다.

10
10 -4 3 1 5 6 -35 12 21 -1

10
2 1 -4 3 4 -4 6 5 -5 1

5
-1 -2 -3 -4 -5

출력

첫째 줄에 답을 출력한다.

33

14

-1

---

이 문제는 연속된 몇 개의 수를 선택해서 구할 수 있는 합중에 가장 큰 합을 구해야한다. 연속되기만 한다면 수의 개수는 한 개 이상이기만 하면 된다.
그래서 입력되는 값과 똑같은 크기의 배열을 하나 만든 다음, 반복문으로 처음부터 끝까지 배열을 훑으면서 이전부터 더해온 부분+현재부분과 현재부분 중에 더 큰 것을 배열에 저장해서 남겨둔다.

Python 풀이

n = int(input())
arr = list(map(int, input().split()))
dp = [0]*len(arr)
dp[0] = arr[0]

for i in range(1, n):
    dp[i] = max(arr[i], dp[i-1]+arr[i])

print(max(dp))

Kotlin 풀이

import java.io.*
import java.util.*

fun main(args: Array<String>)
{
    val br = BufferedReader(InputStreamReader(System.`in`))
    val n = br.readLine().toInt()
    val arr = Array(n){0}
    val dp = Array(n){0}

    val st = StringTokenizer(br.readLine())
    for (i in 0 until n) {
        arr[i] = st.nextToken().toInt()
    }

    dp[0] = arr[0]

    for (i in 1 until n) {
        dp[i] = Math.max(arr[i], dp[i-1]+arr[i])
    }

    println(dp.maxOrNull())
}