LeetCode - 120. 三角形最小路径和

120. 三角形最小路径和

给定一个三角形，找出自顶向下的最小路径和。每一步只能移动到下一行中相邻的结点上。

例如，给定三角形：

[
     [2],
    [3,4],
   [6,5,7],
  [4,1,8,3]
]

自顶向下的最小路径和为 11（即，2 + 3 + 5 + 1 = 11）。

说明：

如果你可以只使用 O(n) 的额外空间（n 为三角形的总行数）来解决这个问题，那么你的算法会很加分。

解题思路： $O(n^2)$空间复杂度的算法应该是非常好想到的，此题属于经典的动态规划题。

class Solution {
public:
    int minimumTotal(vector<vector<int>>& triangle) {
        int n = triangle.size();
        vector<vector<int>> dp = triangle;
        dp[0][0] = triangle[0][0];
        for (int i = 1; i < n; ++i) {
            for (int j = 0; j < triangle[i].size(); ++j) {
                if (j == 0) dp[i][0] = dp[i - 1][0] + triangle[i][0];
                else if (j == triangle[i].size() - 1) {
                    dp[i][j] = dp[i - 1][j - 1] + triangle[i][j];
                } else {
                    dp[i][j] = min(dp[i - 1][j - 1], dp[i - 1][j]) + triangle[i][j];
                }
            }
        }
        int res = INT_MAX;
        for (int j = 0; j <dp[n - 1].size(); ++j) {
            res = min(res, dp[n - 1][j]);
        }
        return res;
    }
};

题目要求空间复杂度为$O(n)$，但是这个算法我没想出来，参考了Grandyang博客，基本思路是定义一个与triangle最后一行相同的一维数组，然后从倒数第二层向第0层扫描，更新每一列的最小值，最终的最小值会汇聚到第0列，因为第0行只有第0列。

class Solution {
public:
    int minimumTotal(vector<vector<int>>& triangle) {
        vector<int> dp(triangle.back());
        for (int i = (int)triangle.size() - 2; i >= 0; --i) {
            for (int j = 0; j <= i; ++j) {
                dp[j] = min(dp[j], dp[j + 1]) + triangle[i][j];
            }
        }
        return dp[0];
    }
};