Algorithm/Baekjoon Online Judge

[다이나믹 프로그래밍] BOJ_1932 숫자삼각형

best 2017. 2. 17. 04:02

문제

        7
      3   8
    8   1   0
  2   7   4   4
4   5   2   6   5

위 그림은 크기가 5인 숫자 삼각형의 한 모습이다.

맨 위층 7부터 시작해서 아래에 있는 수 중 하나를 선택하여 아래층으로 내려올 때, 이제까지 선택된 수의 합이 최대가 되는 경로를 구하는 프로그램을 작성하라. 아래층에 있는 수는 현재 층에서 선택된 수의 대각선 왼쪽 또는 대각선 오른쪽에 있는 것 중에서만 선택할 수 있다.

삼각형의 크기는 1 이상 500 이하이다. 삼각형을 이루고 있는 각 숫자는 모두 정수이며, 범위는 0 이상 99 이하이다.

입력

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

출력

첫째 줄에는 최대가 되는 합을 출력한다.

예제 입력 

5
7
3 8
8 1 0 
2 7 4 4
4 5 2 6 5

예제 출력 

30

코드


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
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
package dp;
 
import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.StringTokenizer;
 
public class baekjoon_1932 {
 
    private void solve() {
        int n = sc.nextInt();
        int[][] dp = new int[n][n];
        
        // 2차원 배열에 담
        for ( int i = 0; i < n; i++ ){
            for (int j = 0; j <= i; j++) {
                dp[i][j] = sc.nextInt();
            }
        }
        
        forint i = 1; i < n; i++ ){
            for ( int j = 0; j <= i; j++ ){
                if( j == 0) { // 왼쪽 끝일 경우 윗줄 왼쪽 끝
                    dp[i][j] = dp[i][j] + dp[i-1][j];
                }
                else if ( j == i ){ // 오른쪽 끝일 경우 윗줄 오른쪽 끝
                    dp[i][j] = dp[i][j] + dp[i-1][j-1];
                }
                else { // 오른쪽, 왼쪽 대각선 모두 비교
                    dp[i][j] = Math.max(dp[i][j] + dp[i-1][j-1], dp[i][j] + dp[i-1][j]);
                }
            }
        }
        
        int max = 0;
        forint i = 0; i < n; i++ ){
            /*for (int j = 0; j <= i; j++){
                System.out.print(dp[i][j] + " ");
            }
            System.out.println("");*/
            if ( max < dp[n-1][i] ){ // 마지막 줄에서 최대값 찾기
                max = dp[n-1][i];
            }
        }
        System.out.println(max);
    }
    
    public static void main(String[] args) {
        // TODO Auto-generated method stub
        sc.init();
        
        new baekjoon_1932().solve();
 
    }
    
    static class sc { 
        private static BufferedReader br;
        private static StringTokenizer st;
        
        static void init() {
            br = new BufferedReader(new InputStreamReader(System.in));
            st = new StringTokenizer("");
        }
        
        static String readLine() {
            try{
                return br.readLine();
            } catch (IOException e){
                e.printStackTrace();
            }
            return null;
        }
        
        static String next() {
            while (!st.hasMoreTokens() ){
                try {
                    st = new StringTokenizer(br.readLine());
                } catch (IOException e){
                    e.printStackTrace();
                }
            }
            return st.nextToken();
        }
        
        static long nextLong() {
            return Long.parseLong(next());
        }
        
        static int nextInt() {
            return Integer.parseInt(next());
        }
        
        static double nextDouble() {
            return Double.parseDouble(next());
        }
        
    }
}
 
cs