다이나믹 프로그래밍

정수 삼각형

4gats 2023. 3. 28. 17:42

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

 

1932번: 정수 삼각형

첫째 줄에 삼각형의 크기 n(1 ≤ n ≤ 500)이 주어지고, 둘째 줄부터 n+1번째 줄까지 정수 삼각형이 주어진다.

www.acmicpc.net

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
46
47
48
49
50
51
#include <iostream>
#include <algorithm>
 
using namespace std;
 
int tri[504][504];
int d[504][504];
 
int main()
{
    ios_base::sync_with_stdio(false);
    cin.tie(nullptr);
 
    int n;
    cin >> n;
 
    for (int i = 1; i <= n; i++)
    {
        for (int j = 1; j <= i; j++)
            cin >> tri[i][j];
    }
 
    d[1][1] = tri[1][1];
    d[2][1] = d[1][1] + tri[2][1];
    d[2][2] = d[1][1] + tri[2][2];
 
    for (int i = 3; i <= n; i++)
    {
        for (int j = 1; j <= i; j++) 
        {
            if (j - 1 == 0)
                d[i][j] = d[i -1][j] + tri[i][j];
            if (j + 1 > n)
                d[i][j] = d[i - 1][j - 1] + tri[i][j];
 
            d[i][j] = max(d[i - 1][j - 1], d[i - 1][j]) + tri[i][j];
        }
    }
    
    int ans = 0;
 
    for (int j = 1; j <= n; j++)
    {
        if (ans < d[n][j])
            ans = d[n][j];
    }
    
    cout << ans << '\n';
    
    return 0;
}
 cs