[백준] 연속합 2(13398번) - 파이썬(Python)

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

13398번: 연속합 2

문제

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보다 작거나 같은 정수이다.

출력

첫째 줄에 답을 출력한다.

예제

Example 1:

Input: 
10
10 -4 3 1 5 6 -35 12 21 -1
Output: 
54

조건

시간 제한 : 2초

메모리 제한 : 512 MB

풀이과정

풀이

import copy
n = int(input())

arr = list(map(int, input().split()))

dp1 = copy.deepcopy(arr)
dp2 = copy.deepcopy(arr)

for i in range(n):
    if i == 0:
        continue
    dp1[i] = max(arr[i], dp1[i-1]+arr[i])

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

ans = max(dp1)

for i in range(1, n-1):
    ans = max((dp1[i-1]+dp2[i+1]), ans)

print(ans)