leetcode-1289 下降路径最小和 II

原题链接：https://leetcode-cn.com/problems/minimum-falling-path-sum-ii/

给你一个整数方阵 arr ，定义「非零偏移下降路径」为：从 arr 数组中的每一行选择一个数字，且按顺序选出来的数字中，相邻数字不在原数组的同一列。

请你返回非零偏移下降路径数字和的最小值。

 

示例 1：

输入：arr = [[1,2,3],[4,5,6],[7,8,9]]
输出：13
解释：
所有非零偏移下降路径包括：
[1,5,9], [1,5,7], [1,6,7], [1,6,8],
[2,4,8], [2,4,9], [2,6,7], [2,6,8],
[3,4,8], [3,4,9], [3,5,7], [3,5,9]
下降路径中数字和最小的是 [1,5,7] ，所以答案是 13 。
 

提示：

1 <= arr.length == arr[i].length <= 200
-99 <= arr[i][j] <= 99

 

分析：相对下降路径最小和问题，本问题的限制是下一行选择的数字和上一行数字不同列，因此可以用dp的思想解这个问题。

首先从上到下从左到右遍历，dp[j]表示第i行第j列第最小值，所以状态转移方程可以写为：

如果上一行的最小值不在第j列，dp[j]=取上一行的最小值+arr[i][j]。

否则dp[j]=取上一行的次小值+arr[i][j]。

所以每次遍历完一行了以后可以算出该行的最小值和次小值以及最小值的坐标index，便于下次遍历。

因此这个方法的时间复杂度上O(N^2),空间复杂度上O(N)

 

class Solution {
    public int minFallingPathSum(int[][] arr) {
        if(arr==null || arr.length==0){
            return 0;
        }
        int[] dp=new int[arr[0].length];
        int mins=0;
        int sub_mins=0;
        int min_index=0;
        for(int i=0;i<arr.length;i++){
            for(int j=0;j<arr[i].length;j++){
                if(i==0){
                    dp[j]=arr[i][j];
                }
                else{
                    if(min_index==j){
                        dp[j]=sub_mins+arr[i][j];
                    }
                    else{
                        dp[j]=mins+arr[i][j];
                    }
                }
            }
            mins=Integer.MAX_VALUE;
            sub_mins=Integer.MAX_VALUE;
            for(int k=0;k<dp.length;k++){
                if(dp[k]<mins){
                    sub_mins=mins;
                    mins=dp[k];
                    min_index=k;
                }
                else if(dp[k]<sub_mins){
                    sub_mins=dp[k];
                }
            }
        }
        return mins;
    }
}