Adam's blog
Published by Adam

[Leetcode]120.三角形路径最小和

---恢复内容开始---

题目的链接

简单的动态规划题,使用了二维dp数组就能很好的表示。

由于有边界的问题,所以这个dp数组为 dp[n+1][n+1]。

dp[i][j]意思是终点为(i-1,j-1)点的路径最小和。

我们需要把这个三角形变成方阵来看,先看看样例:

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

变成方阵之后就变成了

[

[2, INT_MAX,INT_MAX, INT_MAX],

[3,               4,INT_MAX, INT_MAX],

[6,               5,             7, INT_MAX],

[4,               1,             8,              3],

]

有上面方阵很容易得出这个状态转移方程为

dp[i][j]=min(dp[i-1][j-1],dp[i-1][j])+triangle[i-1][j-1];

为了避开数组越界(人i=0或j=0)的问题,我们的dp数组容量比triange大一:即triangle[i][j]->dp[i+1][j+1]

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

或者根本不用再建立一个新的dp数组,而是直接在triangle数组上进行操作。比如

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

这一题的升级版问题可以看我的另一篇随笔: 下降路径最小和

posted @ 2019-01-03 15:35  AdamWong  阅读(206)  评论(0编辑  收藏  举报