2021-03-12

120. 三角形最小路径和

难度中等709

给定一个三角形 triangle ，找出自顶向下的最小路径和。

每一步只能移动到下一行中相邻的结点上。相邻的结点 在这里指的是 下标 与 上一层结点下标 相同或者等于 上一层结点下标 + 1 的两个结点。也就是说，如果正位于当前行的下标 i ，那么下一步可以移动到下一行的下标 i 或 i + 1 。

 

示例 1：

输入：triangle = [[2],[3,4],[6,5,7],[4,1,8,3]]
输出：11
解释：如下面简图所示：
   2
  3 4
 6 5 7
4 1 8 3
自顶向下的最小路径和为 11（即，2 + 3 + 5 + 1 = 11）。

示例 2：

输入：triangle = [[-10]]
输出：-10

 

提示：

• 1 <= triangle.length <= 200
• triangle[0].length == 1
• triangle[i].length == triangle[i - 1].length + 1
• -104 <= triangle[i][j] <= 104

 

进阶：

• 你可以只使用 O(n) 的额外空间（n 为三角形的总行数）来解决这个问题吗？

 

class Solution {
public:
    // vector<vector<int>> triangle;
    // int help(int i, int j, long rs){
    //     if(i==triangle.size()) return rs;
    //     if(j+1<=i){
    //         return min(help(i+1, j, rs+triangle[i][j]), help(i+1, j+1, rs+triangle[i][j+1]));
    //     }
    //     return help(i+1, j, rs+triangle[i][j]);
    // }
    // int minimumTotal(vector<vector<int>>& tri) {
    //     triangle = tri;
    //     return help(0, 0, 0);
    // }
};

 

递归方式最后一个用例会超时，换成二维的动态规划ok：

class Solution {
public:
    int minimumTotal(vector<vector<int>>& tri) {
        int m = tri.size(), n = tri[m-1].size(), dp[m][n];
        memset(dp, 0, sizeof(dp));
        dp[0][0] = tri[0][0];
        for(int i = 1; i < m; i++){
            dp[i][0] = dp[i-1][0] + tri[i][0];
            for(int j = 1; j < i; j++){
                dp[i][j] = min(dp[i-1][j-1], dp[i-1][j]) + tri[i][j];
            }
            dp[i][i] = dp[i-1][i-1] + tri[i][i];
        }
        int rs = dp[m-1][0];
        for(int i = 1; i < n; i++){
            rs = min(dp[m-1][i], rs);
        }
        return rs;
    }
};

路径压缩版本：

class Solution {
public:
    //路径压缩版
    int minimumTotal(vector<vector<int>>& tri) {
        int m = tri.size(), n = tri[m-1].size(), dp[m];
        memset(dp, 0, sizeof(dp));
        dp[0] = tri[0][0];
        for(int i = 1; i < m; i++){
            dp[i] = dp[i-1] + tri[i][i];
            for(int j = i-1; j > 0; j--){
                dp[j] = min(dp[j-1], dp[j]) + tri[i][j];
            }
            dp[0] = dp[0] + tri[i][0];
        }
        int rs = dp[0];
        for(int i = 1; i < n; i++){
            rs = min(dp[i], rs);
        }
        return rs;
    }
};