动态规划 每日题目 力扣题目7

作者：宫水三叶
链接：https://leetcode.cn/leetbook/read/path-problems-in-dynamic-programming/r85adr/
来源：力扣（LeetCode）
著作权归作者所有。商业转载请联系作者获得授权，非商业转载请注明出处。

解析

本题能够从首行的任意位置开始转移，定义一个$ find()$ 函数，传入 矩阵 和 首行的起点下标，返回「以首行该下标」为起点的 最小路径和。
即所有$ find(matrix,i)$ 的最小值，首先需要枚举起点，在进行DP，这样的复杂度是$O(n^3)$
对于一个确定的起点，找到其「最小路径和」的路径需要转移$O(n^2)$个状态。

class Solution {
public:
    


    int find(vector<vector<int>>& matrix,int column){
        //find the minimum value that begin with i
        int n = matrix.size();
        int f[n][n];
       //f[0][0]=matrix[0][column];//start at row 0 column i
        for(int i=0;i<n;i++){
            f[0][i] = i==column?matrix[0][i]:INT_MAX;
        }
        for(int i=1;i<n;i++){
            for(int j=0;j<n;j++){
                f[i][j]=INT_MAX;
                if(f[i-1][j]!=INT_MAX)
                    f[i][j]=min(f[i-1][j]+matrix[i][j],f[i][j]);
                if(j>=1&&f[i-1][j-1]!=INT_MAX){
                    f[i][j]=min(f[i][j],f[i-1][j-1]+matrix[i][j]);
                }
                if(j<n-1&&f[i-1][j+1]!=INT_MAX){
                    f[i][j]=min(f[i][j],f[i-1][j+1]+matrix[i][j]);
                }
                    
                
            }

        }
        int ans=INT_MAX;
        for(int i=0;i<n;i++) ans = min(ans,f[n-1][i]);
        return ans;

    }

    int minFallingPathSum(vector<vector<int>>& matrix) {
        int n = matrix.size();
        int max = INT_MAX;
        for(int i=0;i<n;i++){
            max = min(max,find(matrix,i));
        }
        return max;
        
        
    }
};

优化

上述做法分为两步：1. 枚举起点 2. DP求每个起点的最小路径和。
DP过程的复杂度没办法优化，枚举起点的过程可以优化。直接从DP定义出发进行转移。
定义$f[i][j]$为到位置$(i,j)$的最小值，$i$ 的取值范围$[0,n)$。最小路径的结尾可能是最后一行任意位置。

class Solution {
public:
    

    int minFallingPathSum(vector<vector<int>>& matrix) {
        int n = matrix.size();
        int max = INT_MAX;
        int f[n][n];

        for(int i=0;i<n;i++){
            f[0][i] = matrix[0][i];
        }
        for (int i = 1; i < n; i++) {
            for (int j = 0; j < n; j++) {
                f[i][j] = f[i-1][j]+matrix[i][j];
                if(j>0) 
                    f[i][j] = min(f[i][j], f[i-1][j-1]+matrix[i][j]);
                if(j<n-1) 
                    f[i][j] = min(f[i][j],f[i-1][j+1]+matrix[i][j]);
            }
        }
        for (int i = 0; i < n; i++) max = min(max,f[n-1][i]);

        return max;
        
        
    }
};