1 분 소요

문제

        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번째 줄까지 정수 삼각형이 주어진다.

출력

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


접근방법

삼각형의 아래층부터 시작해서 위로 더해가는 방식으로 풀었다.

가장 아래층에 있는 값을 그대로 배열 d에 넣어주고,

현재의 값과 그 오른쪽에 있는 값 중에 최댓값을 위층에 있는 값과 더해서 배열 d에 저장해주었다.

계속 동일한 방식으로 더해가다보면 최종적으로 제일 위층에 아래층에서부터 누적된 최댓값이 들어가게 된다.

풀이

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.*;

public class Main {
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        int n = Integer.parseInt(br.readLine());
        int[][] triangle = new int[n][n];
        int[][] d = new int[n][n];
        
        for(int i = 0; i < n; i++) {
            StringTokenizer st = new StringTokenizer(br.readLine());
            for(int j = 0; j < i+1; j++) {
                triangle[i][j] = Integer.parseInt(st.nextToken());
            }
        }
        
        for(int i = 0; i < n; i++) {
            d[n-1][i] = triangle[n-1][i];
        }
        
        for(int i = 2; i <= n; i++) {
            for(int j = 0; j <= n-i; j++) {
                d[n-i][j] = Math.max(d[n-i+1][j], d[n-i+1][j+1]) + triangle[n-i][j];
            }
        }
        
        System.out.println(d[0][0]);
    }
}

댓글남기기