Leetcode 931 Minimum Falling Path Sum

问题描述

给定一个方阵A，我们想要得到A的最小下降路径元素之和。下降路径是指从第一行开始，每一行都选择一个元素，但是下一行选择的元素必须是和前一行最多相隔一列。

例子

输入: [[1,2,3],[4,5,6],[7,8,9]]
输出: 12
解释: 
那么可能的下降路径为:

 - [1,4,7], [1,4,8], [1,5,7], [1,5,8], [1,5,9]
 - [2,4,7], [2,4,8], [2,5,7], [2,5,8], [2,5,9], [2,6,8], [2,6,9]
 - [3,5,7], [3,5,8], [3,5,9], [3,6,8], [3,6,9]
最小下降路径为 [1,4,7],和为12.

解析

暴力破解显然不合适，指数级的搜索开销。这是一个动态规划问题，我们从n=1开始归纳，即是A[0][0];当n=2时,最小下降路径为min(dp[0][0],dp[0][1])，其中dp[i][j]表示从A[i][j]开始下路到最后一层的最小值。更一般的，dp[i][j]=A[]i[j]+min(dp[i+1][j-1], dp[i+1][j],dp[i+1][j+1])。

Java代码

    public int minFallingPathSum(int[][] A) {
        int m = A.length;
        int dp[][] = new int[m][m];
        for(int i = 0; i < m; i++){
            dp[m-1][i] = A[m-1][i];
        }
        for(int i=m-2;i>=0;i--){
            for(int j=0;j<m;j++){
                if(j-1<0){
                    dp[i][j] = A[i][j] + Math.min(dp[i+1][j], dp[i+1][j+1]);
                }
                else if(j + 1>m-1){
                    dp[i][j] = A[i][j] + Math.min(dp[i+1][j], dp[i+1][j-1]);
                }
                else{
                    dp[i][j] = A[i][j] + Math.min(Math.min(dp[i+1][j-1], dp[i+1][j]), dp[i+1][j+1]);
                }
                
            }
        }
        int result=dp[0][0];
        for(int i: dp[0]){
            result = Math.min(result, i);
        }
        return result;
    }