leetcode 120. 三角形最小路径和

leetcode 120. 三角形最小路径和

题目详情

题目链接
给定一个三角形，找出自顶向下的最小路径和。每一步只能移动到下一行中相邻的结点上。
相邻的结点 在这里指的是 下标 与 上一层结点下标 相同或者等于 上一层结点下标 + 1 的两个结点。
例如，给定三角形：
[
[2],
[3,4],
[6,5,7],
[4,1,8,3]
]
自顶向下的最小路径和为 11（即，2 + 3 + 5 + 1 = 11）。
说明：

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

我的代码

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

我的成绩

执行结果：通过
执行用时：12 ms, 在所有 C++ 提交中击败了48.21%的用户
内存消耗：8.2 MB, 在所有 C++ 提交中击败了100.00%的用户

一些想法

本道题是一道简单的动态规划题，直接在原数组上进行每个位置的更新即可。

执行用时为 0 ms 的范例

class Solution {
public:
    int minimumTotal(vector<vector<int>>& tri) {
        int n = tri.size(),m = tri[n - 1].size();

        vector<int> f[2];
        f[0] = f[1] = vector<int> (n);
        f[0][0] = tri[0][0];
        for(int i = 1 ; i < n ;i++)
            for(int j = 0 ; j <= i ;j++)
            {
                vector<int> &a = f[i & 1],&b = f[i - 1 & 1];
                a[j] = INT_MAX;
                if(j > 0)a[j] = b[j - 1] + tri[i][j];
                if(j < i)a[j] = min(a[j],b[j] + tri[i][j]);
            }
        int res = INT_MAX;
        for(int i = 0 ; i < n ;i++)res = min(res,f[n - 1 & 1][i]);
        return res;
    }
};

思考

方法类似，参见官方解答