PS/Java

[백준] 1932. 정수 삼각형

siyamaki 2022. 4. 6. 13:46
        7
      3   8
    8   1   0
  2   7   4   4
4   5   2   6   5

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

 

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

 

아래층에 있는 수는 현재 층에서 선택된 수의 대각선 왼쪽 또는 대각선 오른쪽에 있는 것 중에서만 선택할 수 있다.

 

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


입력

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

출력

첫째 줄에 합이 최대가 되는 경로에 있는 수의 합을 출력한다.


예제 입력 1

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

예제 출력 1

30

import java.io.BufferedReader;
import java.io.InputStreamReader;
import java.util.StringTokenizer;

public class Main {
    public static int N;
    public static int[] arr;
    public static void main(String[] args) throws Exception {

        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        N = Integer.parseInt(br.readLine());
        arr = new int[sumN(N)];
        int idx = 1;
        int max = Integer.MIN_VALUE;
        arr[0] = 0;

        for(int i = 1; i <= N; i++) {
            StringTokenizer st = new StringTokenizer(br.readLine());

            for(int j = 1; j <= i; j++) {
                int num = Integer.parseInt(st.nextToken());
                if(j == 1) {
                    arr[idx] = arr[idx - (i - 1)] + num;
                } else if (j == i) {
                    arr[idx] = arr[idx - i] + num;
                } else {
                    // 사이값 비교
                    if(arr[idx - i] > arr[idx - i + 1]) {
                        arr[idx] = arr[idx - i] + num;
                    } else {
                        arr[idx] = arr[idx - i + 1] + num;
                    }
                }
                
                if(arr[idx] >= max) {
                    max = arr[idx];
                }
                idx++;
            }

        }
        
        System.out.println(max);
    }

    public static int sumN(int n) {
        int sum = 0;
        for(int i = 1; i <= n; i++) {
            sum += i;
        }

        return sum + 1;
    }
}

자신의 현재 위치로부터 다음 Index를 접근했을 때 어느 값이 더 큰지 비교하면서 아래로 내려간다. 다음 인덱스로 갈 때 이전 값을 arr배열에 계속 저장해나간다.

 

                0
                1
              2   3
            4   5   6
          7   8   9  10
        11  12  13  14  15
 해당 삼각형의 Index의 규칙을 찾아서 접근한다.