[코테] 백준 다이나믹 프로그래밍 13398번 문제

13398번 : 연속합 2

문제 링크

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

문제 설명

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 ≤ 500)이 주어지고, 둘째 줄부터 n+1번째 줄까지 정수 삼각형이 주

출력

첫째 줄에 답을 출력한다.

문제풀이

dp 배열을 2개 선언하는데 첫번째는 그냥 연속합을 구하는 것이고, 두번째는 수 한개를 제외한 연속합을 구하고, 이중에서 최대값을 구한다.

import sys
input = sys.stdin.readline

N = int(input())
arr = list(map(int, input().split()))
result = arr[0]

dp = [arr.copy() for _ in range(2)]
dp[1][0] = 0


for i in range(1, N) : 
    dp[0][i] = max(dp[0][i-1] + dp[0][i], dp[0][i])
    dp[1][i] = max(dp[0][i-1], dp[1][i-1] + dp[1][i])
    result = max(result, max(dp[0][i], dp[1][i]))

print(result)