Always Be Wise

동적 프로그래밍 : 연속합 2(백준 13398번) 본문

알고리즘/백준

동적 프로그래밍 : 연속합 2(백준 13398번)

bewisesh91 2021. 12. 1. 20:07
728x90

▶ 문제 : https://www.acmicpc.net/problem/13398

 

13398번: 연속합 2

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

www.acmicpc.net

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

##### 출력 #####
# 첫째 줄에 답을 출력한다.

 

▶ 접근 방법

기존 연속합을 구하는 문제에서 수를 제거할 수 있다는 조건이 들어간 변형 문제였다.
DP 테이블을 수를 제거하지 않은 경우와, 수를 제거한 경우로 구분하여 만들고
해당 테이블 마다의 점화식을 세워 문제를 해결하였다.

 

▶ 풀이 코드

import sys

N = int(sys.stdin.readline().rstrip())
numbers = list(map(int, sys.stdin.readline().split()))

DP = [[0] * (N + 1) for _ in range(2)]

DP[0][0] = numbers[0]
DP[1][0] = 0

result = numbers[0]
for idx in range(1, N) :
    # 하나도 제거하지 않은 경우
    DP[0][idx] = max(DP[0][idx-1] + numbers[idx], numbers[idx])
    # 하나 제거한 경우
    DP[1][idx] = max(DP[1][idx-1] + numbers[idx], DP[0][idx-1])
    result = max(result, DP[0][idx], DP[1][idx])

print(result)

 

▶ 관련 링크

2021.11.25 - [알고리즘] - 동적 프로그래밍(Dynamic Programming)이란?

2021.11.25 - [알고리즘] - 동적 프로그래밍 : 피보나치 수 2(백준 2748번)

2021.11.25 - [알고리즘] - 동적 프로그래밍 : 01타일(백준 1904번)

2021.11.26 - [알고리즘] - 동적 프로그래밍 : 평범한 배낭(백준 12865번)

2021.11.27 - [알고리즘] - 동적 프로그래밍 : 동전(백준 9084번)

2021.11.27 - [알고리즘] - 동적 프로그래밍 : LCS(백준 9251번)

2021.11.27 - [알고리즘] - 동적 프로그래밍 : 가장 긴 증가하는 부분 수열(백준 11053번)

2021.11.29 - [알고리즘] - 동적 프로그래밍 : 점프(백준 2253번)

2021.11.30 - [알고리즘] - 동적 프로그래밍 : 연속합(백준 1912번)

2021.12.01 - [알고리즘] - 동적 프로그래밍 : 계단 오르기(백준 2579번)

2021.12.01 - [알고리즘] - 동적 프로그래밍 : RGB거리(백준 1149번)

2021.12.01 - [알고리즘] - 동적 프로그래밍 : 포도주 시식(백준 2156번)

2021.12.01 - [알고리즘] - 동적 프로그래밍 : 이동하기(백준 11048번)

Comments