[DP] 프로그래머스 정수삼각형 "JAVA"

🚀링크

https://programmers.co.kr/learn/courses/30/lessons/43105?language=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 배열에 넣어주면서 내려온다.

경우의 수는
1) 현 위치가 가장 왼쪽 값인 경우
위 트리로부터 가져올 수 있는 값은 DP[i-1][j] 밖에 없으니 DP[i-1][j] + triangle[i][j]
2) 현 위치가 가장 오른쪽 값인 경우
위 트리로부터 가져올 수 있는 값은 DP[i-1][j-1] 밖에 없으니 DP[i-1][j-1] + triangle[i][j]
3) 그 외의 경우
현 위치로 부터 위 트리의 왼쪽 값의 합과 오른쪽 값 중에 큰 값을 가져와야 함
따라서 Math.max(DP[i-1][j-1], DP[i-1][j]) + triangle[i][j]

👩‍💻코드

class Solution {
    public int solution(int[][] triangle) {
        int answer = 0;
        int dp[][] = new int[triangle.length][triangle.length];
        
        dp[0][0] = triangle[0][0];
        
        for(int i=1; i<triangle.length; i++){
            for(int j=0; j<triangle[i].length; j++){
                if(j==0){//맨 왼쪽
                    dp[i][j] = dp[i-1][j] + triangle[i][j];
                }else if(j==triangle[i].length-1){//맨 오른쪽
                    dp[i][j] = dp[i-1][j-1] + triangle[i][j];
                }else{//그 외 경우
                    dp[i][j] = Math.max(dp[i-1][j-1], dp[i-1][j]) + triangle[i][j];
                }
            }
        }
        
        int len = dp.length;
        for(int i=0; i<len; i++){
            answer = Math.max(answer, dp[len-1][i]);
        }
        
        return answer;
    }
}