题目
给定一个三角形,找出自顶向下的最小路径和。每一步只能移动到下一行中相邻的结点上。
相邻的结点 在这里指的是 下标 与 上一层结点下标 相同或者等于 上一层结点下标 + 1 的两个结点。
示例1
输入:
[
[2],
[3,4],
[6,5,7],
[4,1,8,3]
]
输出:
11
说明:
自顶向下的最小路径和为 11(即,2 + 3 + 5 + 1 = 11)。
解法
- 这是一道很中规中矩的动态规划,并且可以只开辟一维数组保存上一次的值,来优化空间。
- 动态转移方程:dp[j] = min(dp[j], dp[j-1]) + triangle[i][j]
- 需要注意矩阵的零界点
代码
#include <stdio.h>
#include <vector>
#include <iostream>
using namespace std;
class Solution {
public:
int minimumTotal(vector<vector<int>>& triangle) {
int n = triangle.size();
vector<int> dp(n);
dp[0] = triangle[0][0];
for (int i = 1; i < n; i++) {
dp[i] = dp[i - 1] + triangle[i][i];
//注意这里的顺序,j=i-1开始减小到1为止,类似于背包问题。
//如果j从1开始,我们就会用已经更新好的值去更新下一个,而我们需要的是上一次的值去更新,造成错误,倒着来能避免冲突
for (int j = i - 1; j > 0; j--) {
dp[j] = min(dp[j], dp[j-1]) + triangle[i][j];
}
dp[0] += triangle[i][0];
}
int minn = dp[0];
for (int i = 1;i < n;i++){
if (minn > dp[i]) {
minn = dp[i];
}
}
return minn;
}
};
int main()
{
vector<vector<int>> m(4,vector<int> (4));
m[0][0] = 2;
m[1][0] = 3;m[1][1] = 4;
m[2][0] = 6;m[2][1] = 5;m[2][2] = 7;
m[3][0] = 4;m[3][1] = 1;m[3][2] = 8;m[3][3] = 3;
Solution s ;
cout<<s.minimumTotal(m);
return 0;
}
今天也是爱zz的一天哦!