題目描述:
給定一個三角形,找出自頂向下的最小路徑和。每一步只能移動到下一行中相鄰的結點上。
例如,給定三角形:
[
[2],
[3,4],
[6,5,7],
[4,1,8,3]
]
自頂向下的最小路徑和爲 11(即,2 + 3 + 5 + 1 = 11)。
說明:
如果你可以只使用 O(n) 的額外空間(n 爲三角形的總行數)來解決這個問題,那麼你的算法會很加分。
解法1:二維dp, 自頂向下,時間複雜度和空間複雜度都是O(n^2)
class Solution {
public:
int minimumTotal(vector<vector<int>>& triangle) {
if(triangle.size()<=0) {
return 0;
}
vector<vector<int>> dp;
for (int i = 0; i < triangle.size(); i++)
{
vector<int> temp;
for (int j = 0; j < triangle[i].size(); j++)
{
if(i==0) {
temp.push_back(triangle[i][0]);
} else if(j == 0) {
temp.push_back(dp[i-1][j]+triangle[i][j]);
} else if(j ==triangle[i].size()-1) {
temp.push_back(dp[i-1][j-1]+triangle[i][j]);
} else {
temp.push_back(min(dp[i-1][j],dp[i-1][j-1])+triangle[i][j]);
}
}
dp.push_back(temp);
}
int minPath=INT32_MAX;
int len=dp.size();
for (int i = 0; i < dp[len-1].size(); i++)
{
if(dp[len-1][i]<minPath) {
minPath=dp[len-1][i];
}
}
return minPath;
}
};
解法2:動態規劃,自底向上,時間複雜度O(n^2), 空間複雜度O(n).
class Solution {
public:
int minimumTotal(vector<vector<int>>& triangle) {
if(triangle.size()<=0) {
return 0;
}
vector<int> dp=triangle[triangle.size()-1];
for (int i = triangle.size()-2; i >=0; i--)
{
for (int j = 0; j < triangle[i].size(); j++)
{
dp[j]=min(dp[j],dp[j+1])+triangle[i][j];
}
}
return dp[0];
}
};