码迷,mamicode.com
首页 > 其他好文 > 详细

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

时间:2019-01-03 16:45:31      阅读:148      评论:0      收藏:0      [点我收藏+]

标签:随笔   https   href   size   log   block   ini   好的   恢复   

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

题目的链接

简单的动态规划题,使用了二维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];
    }
};

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

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

标签:随笔   https   href   size   log   block   ini   好的   恢复   

原文地址:https://www.cnblogs.com/adamwong/p/10214509.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!