코딩고치

[백준][DP] 연속합2 본문

백준 알고리즘 기초/다이내믹 프로그래밍

[백준][DP] 연속합2

코딩고치 2020. 3. 30. 22:20

1. 문제 주소

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

 

13398번: 연속합 2

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

www.acmicpc.net

2. 문제

left_dp[i]에는 0번 항부터 i번 항까지의 수를 더한 값을 입력을 하였고 right_dp[i]에는 i번 항부터 마지막 항까지의 수를 더한 값을 입력을 해주었다. 그리고 left_dp[i-1] + right_dp[i+1]은 i번째 수를 제외한 합이 되므로 이를 이용하여 최댓값을 구하는 코드를 작성하였다.

3. 소스코드

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;
 
int main() {
    int n;
    cin >> n;
 
    vector<int> series(n);
    for (int i = 0; i < n; i++) {
        cin >> series[i];
    }
 
    vector<int> left_dp(n);
    vector<int> right_dp(n);
 
    for (int i = 0; i < n; i++) {
        left_dp[i] = series[i];
        if (i > 0 && left_dp[i] < left_dp[i - 1+ series[i])
            left_dp[i] = left_dp[i - 1+ series[i];
    }
 
    for (int i = n - 1; i >= 0; i--) {
        right_dp[i] = series[i];
        if (i < n - 1 && right_dp[i] < right_dp[i + 1+ series[i])
            right_dp[i] = right_dp[i + 1+ series[i];
    }
 
    int max_value = left_dp[0];
    for (int i = 1; i < n; i++) {
        max_value = max(max_value, left_dp[i]);
    }
    for (int i = 1; i < n - 1; i++) {
        max_value = max(max_value, left_dp[i - 1+ right_dp[i + 1]);
    }
    cout << max_value << '\n';
    return 0;    
}
http://colorscripter.com/info#e" target="_blank" style="color:#4f4f4ftext-decoration:none">Colored by Color Scripter
 
Comments