[LeetCode] 931번 Minimum Falling Path Sum 문제를 풀어보았다. (ft. java)

2022년 12월 13일 화요일 - 이제는 자동화가 잘 붙었어!

오늘 올려볼 문제는 931번 Minimum Falling Path Sum 이라는 문제이다.

사진을 클릭하면 해당 문제로 이동합니다.

오늘도 LeetCode 사이트 오늘의 문제를 가지고 왔다.

조금만 생각했다면 섹시하게 풀 수 있었을 텐데...

입력

사진에서도 볼 수 있듯이 2차원 int 배열 1개가 입력으로 들어온다.

---

풀이 및 코드

위 2차원 배열에서 세로로 합하면서 이었을 때 가장 작은 값이 무엇인지 구하는 문제이다.

오늘은 처음부터 정답을 생각해냈다.

그냥 뭐 딱봐도 dp로 풀면되는 문제이다.

이제 코드를 봐보자!

풀이코드

class Solution {
    public int minFallingPathSum(int[][] matrix) {
        int c = matrix.length, r = matrix[0].length;
        for(int i = 1; i < c; i++) {
            for(int j = 0; j < r; j++) {
                int num = Integer.MAX_VALUE;
                for(int k = -1; k <= 1; k++) {
                    if(0 <= j + k && j + k < r) num = Math.min(num, matrix[i][j] + matrix[i - 1][j + k]);
                }
                matrix[i][j] = num;
            }
        }
        
        int result = matrix[c - 1][0];
        for(int i = 1; i < r; i++) result = Math.min(result, matrix[c - 1][i]);
        
        return result;
    }
}

---

제출 화면

오늘 edge case 때문에 코드가 조금 더러워졌는데 이를 Math.max나 Math.min을 사용하면 훨씬 예쁘게 짤 수 있었다...

다음에는 이런일이 있을 때 꼭 써먹어야겠다!

내일도 문제를 풀어서 블로그에 글을 쓸 수 있으면 좋겠다.