다이나믹 프로그래밍

연속합 2 (중요!)

4gats 2023. 3. 30. 20:58

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

 

13398번: 연속합 2

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

www.acmicpc.net

숫자를 빼고 그 때마다 연속합을 구하면 시간 초과 날 게 뻔하다!!

숫자를 기준으로 나누자!

d[i] = i번째에서 끝나는 연속합

d2[i] = i번째에서 시작하는 연속합

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
40
41
42
43
44
45
#include <iostream>
#include <algorithm>
 
int a[100004];
int d[100004];
int d2[100004];
 
using namespace std;
 
int main()
{
    int n;
    cin >> n;
 
    for (int i = 1; i <= n; i++)
        cin >> a[i];
 
    d[1= a[1];
    for (int i = 2; i <= n; i++)
    {
        d[i] = max(d[i - 1+ a[i], a[i]);
    }
 
    d2[n] = a[n];
    for (int i = n-1; i >= 1; i--)
    {
        d2[i] = max(d2[i + 1+ a[i], a[i]);
    }
//수를 제거하지 않아도 되므로!
    int ans = d[1];
    for (int i = 1; i <= n; i++)
    {
        if (ans < d[i])
            ans = d[i];
    }
 
    // 뺀 거 비교
    for (int i = 2; i <= n - 1; i++)
    {
        if (ans < d[i - 1+ d2[i + 1])
            ans = d[i - 1+ d2[i + 1];
    }
 
    cout << ans << '\n';
}
cs

처음이랑 마지막 빼는 것은 카운트를 하지않는다!

왜? d에서 커버가 된다

합이 최대가 되게 구하고 있으므로 d에서 커버를 했는데

빼는 것은 의미가 없다