源码:
我第一眼以为是搜索法,结果直接超时,而且是最后一个用例过不去。
class Solution {
public:
int result = INT_MAX;
void help(vector<vector<int>>& triangle, int row, int col, int tmp){
if(row == triangle.size()){
result = min(result, tmp);
return;
}
tmp += triangle[row][col];
help(triangle, row+1, col, tmp);
help(triangle, row+1, col+1, tmp);
}
int minimumTotal(vector<vector<int>>& triangle) {
if(triangle.size() == 0) return 0;
help(triangle, 0, 0, 0);
return result;
}
};
后来想想这样重复的太多了,比如题目例子中的能到5的点有两个(3,4),但是不一定要全部都遍历。选一个即可。但是这样又要保存中间的变量,这不就是动态规划吗。效率很高。
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, 0));
dp[0][0] = triangle[0][0];
for(int i=1; i<n; i++){
dp[i][0] = triangle[i][0]+dp[i-1][0];
for(int j=1; j<i; j++){
dp[i][j] = min(dp[i-1][j-1], dp[i-1][j])+triangle[i][j];
}
dp[i][i] = triangle[i][i]+dp[i-1][i-1];
}
int result = INT_MAX;
for(int i=0; i<n; i++){
result = min(result, dp[n-1][i]);
}
return result;
}
};
还有自底向上的一维动态规划(参考了别人):
class Solution {
public:
int minimumTotal(vector<vector<int>>& triangle){
int len = triangle.size();
vector<int> dp(len,0);
for(int i = len - 1; i >= 0; --i){
for(int j = 0; j <= i; ++j){
if(i == len - 1) dp[j] = triangle[i][j];
else dp[j] = triangle[i][j] + min(dp[j],dp[j+1]);
}
}
return dp[0];
}
};