题目来自leetcode上的Array类的习题,题目如下。
题目链接:https://leetcode.com/problems/triangle/description/
这是一道难度中等的题目,leetcode上的通过率为36%。
题目要求我们寻找从三角形的顶端到底部的最短路径,即pick到的数之和最小。题目规定每次只能移动到adjacent numbers on the row below,翻译过来即下一行的相邻数字,但是我按照这个规则实现的算法发现是wrong answer,查看它给的标准答案发现,其实移动的规则应该是 每次只能移动到下一行的相同位置的数字或者右边相邻的数字,不能移动到左边相邻的数字,所以这里是有个坑的。
说回解题思路,(这里只讲我当时做题的思路,并不是最优解,更优的解请去看leetcode上高质量的discussion)因为自上到下选择不同的数字相加,结果会影响到下一步的选择,导致最后的结果不同。也就是说下一步的选择会用到前一步的问题的解,这符号DP算法问题的重叠子问题性质,同时后续的决策不会影响前面的结果,又满足无后续性质。所以这个问题可以使用DP算法解决。
用一个二维数组来保存每个阶段的状态,外层数组存的是移动的次数,内层数组存的是状态。
状态转移方程如下:
if(clo ==0)
dp[i][clo] = triangle[i][clo] + dp[i-1][0];
else
dp[i][clo] = min(dp[i-1][clo-1], dp[i-1][clo]) + triangle[i][clo];
具体代码:
#include <iostream>
#include <algorithm>
#include <vector>
using namespace std;
class Solution {
public:
int minimumTotal(vector<vector<int>>& triangle) {
int row;
int mininum;
row = triangle.size();
vector<vector<int> > dp(row, vector<int>(row+1, INT_MAX));
dp[0][0] = triangle[0][0];
for(int i=1; i<row; i++) {
for(int clo = 0; clo <= i; clo++) {
if(clo ==0)
dp[i][clo] = triangle[i][clo] + dp[i-1][0];
else
dp[i][clo] = min(dp[i-1][clo-1], dp[i-1][clo]) + triangle[i][clo];
}
}
mininum = dp[row-1][0];
for(int i = 1; i < triangle[row-1].size();i++) {
mininum = min(mininum, dp[row-1][i]);
}
return mininum;
}
};