給定一個三角形,找出自頂向下的最小路徑和。每一步只能移動到下一行中相鄰的結(jié)點上。
相鄰的結(jié)點 在這里指的是 下標(biāo) 與 上一層結(jié)點下標(biāo) 相同或者等于 上一層結(jié)點下標(biāo) + 1 的兩個結(jié)點。
例如,給定三角形:
[
[2],
[3,4],
[6,5,7],
[4,1,8,3]
]
自頂向下的最小路徑和為 11(即,2 + 3 + 5 + 1 = 11)。
說明:
如果你可以只使用 O(n) 的額外空間(n 為三角形的總行數(shù))來解決這個問題,那么你的算法會很加分。
class Solution {
public:
int minimumTotal(vector<vector<int>>& triangle) {
int n = triangle.size();
if (n == 0)
return 0;
vector<vector<int>> dp(n, vector<int>(n));
dp[0][0] = triangle[0][0];
for (int i = 1; i < n; i++) {
dp[i][0] = triangle[i][0] + dp[i-1][0];
dp[i][i] = triangle[i][i] + dp[i-1][i-1];
for (int j = 1; j < i; j++)
dp[i][j] = triangle[i][j] + std::min(dp[i-1][j], dp[i-1][j-1]);
}
return *min_element(dp[n-1].begin(), dp[n-1].end());
}
};
在上面的基礎(chǔ)上,可以對內(nèi)存做進(jìn)一步優(yōu)化,直接把dp值累加到三角形中
class Solution {
public:
int minimumTotal(vector<vector<int>>& triangle) {
int n = triangle.size();
if (n == 0)
return 0;
for (int i = 1; i < n; i++) {
triangle[i][0] += triangle[i-1][0];
triangle[i][i] += triangle[i-1][i-1];
for (int j = 1; j < i; j++)
triangle[i][j] += std::min(triangle[i-1][j], triangle[i-1][j-1]);
}
return *min_element(triangle[n-1].begin(), triangle[n-1].end());
}
};
其它的思路:將三角形倒過來計算dp值
class Solution {
public:
int minimumTotal(vector<vector<int>>& triangle) {
int n = triangle.size();
for (int i = n-2; i >= 0; i--) {
int k = triangle[i].size();
for (int j = 0; j < k; j++) {
triangle[i][j] += min(triangle[i+1][j], triangle[i+1][j+1]);
}
}
return triangle[0][0];
}
};