프로그래머스 - 정수 삼각형(Java, 자바)

문제 링크

문제 설명

위와 같은 삼각형의 꼭대기에서 바닥까지 이어지는 경로 중, 거쳐간 숫자의 합이 가장 큰 경우를 찾아보려고 합니다. 아래 칸으로 이동할 때는 대각선 방향으로 한 칸 오른쪽 또는 왼쪽으로만 이동 가능합니다. 예를 들어 3에서는 그 아래칸의 8 또는 1로만 이동이 가능합니다.

삼각형의 정보가 담긴 배열 triangle이 매개변수로 주어질 때, 거쳐간 숫자의 최댓값을 return 하도록 solution 함수를 완성하세요.

제한사항

• 삼각형의 높이는 1 이상 500 이하입니다.
• 삼각형을 이루고 있는 숫자는 0 이상 9,999 이하의 정수입니다.

입출력 예

triangle	result
[[7], [3, 8], [8, 1, 0], [2, 7, 4, 4], [4, 5, 2, 6, 5]]	30

 

---

문제 풀이 방법

문제 풀이 방법은 DP 방법으로 해결하면 되며, 맨왼쪽, 중간, 맨오른쪽 3단계로 나눠서 연산을 하면 된다.

i = 1 행을 보면 양쪽 사이드 값 외에 중간값이 없다.
이때는 맨왼쪽 값과 맨 오른쪽 값은 바로 위의 값에다가 현재값을 누적하면 된다.

i = 2 행에서는 중간 값이 있다.
이때는 바로 위의 값과 왼쪽 대각선의 값을 비교해서 큰 값을 선택한 뒤 현재 값과 더하면 된다.

i = 3 행도 바로 위에서 했던것과 동일하게 반복하면 된다.

마지막으로 i = 4 행도 바로 위에서 했던것과 동일하게 반복하면 된다.

최종적으로 바로 위의 값과 동일한 상태가 된다.
여기서 마지막 행에서 가장 큰 값을 찾아서 return 하면 된다.

맨 왼쪽값은 dp[i][0] = dp[i - 1][0] + triangle[i][0] 이 되며, 맨 오른쪽값은 dp[i][i] = dp[i - 1][i - 1] + triangle[i][i]이 된다.

for (int j = 1; j <= i ; j++) {
    dp[i][j] = Math.max(dp[i - 1][j], dp[i - 1][j - 1]) + triangle[i][j];
}

중간값은 반복문으로 돌리며, Math.max() 을 통해서 큰 값을 구한 다음 누적하면 된다.

Arrays.stream(dp[triangle.length - 1]).max().getAsInt()

최종적으로 구한 배열에서 큰값을 구하는 방법은 반복문으로 큰값을 찾아서 계속 업데이트 하는 방법이 있지만, stream 을 통해서 간단하게 구현 할 수 있다.

---

전체 코드

import java.util.*;
class Solution {
    public int solution(int[][] triangle) {
        int[][] dp = new int[triangle.length][triangle.length];

        dp[0][0] = triangle[0][0];

        for (int i = 1; i < triangle.length; i++) {
            dp[i][0] = dp[i - 1][0] + triangle[i][0];

            for (int j = 1; j <= i ; j++) {
                dp[i][j] = Math.max(dp[i - 1][j], dp[i - 1][j - 1]) + triangle[i][j];
            }

            dp[i][i] = dp[i - 1][i - 1] + triangle[i][i];
        }

        return Arrays.stream(dp[triangle.length - 1]).max().getAsInt();
    }
}