120. 三角形最小路径和
给定一个三角形 triangle ,找出自顶向下的最小路径和。
每一步只能移动到下一行中相邻的结点上。相邻的结点 在这里指的是 下标 与 上一层结点下标 相同或者等于 上一层结点下标 + 1 的两个结点。也就是说,如果正位于当前行的下标 i ,那么下一步可以移动到下一行的下标 i 或 i + 1 。
示例 :
输入:triangle = [[2],[3,4],[6,5,7],[4,1,8,3]]
输出:11
解释:如下面简图所示:
2
3 4
6 5 7
4 1 8 3
自顶向下的最小路径和为 11(即,2 + 3 + 5 + 1 = 11)。
本题采用动态规划的方法。
假设到i行第j个数字的最小路径为f[i][j],则有:
f
[
i
]
[
j
]
=
m
i
n
(
f
[
i
−
1
]
[
j
]
,
f
[
i
−
1
]
[
j
−
1
]
)
+
t
r
i
a
n
g
l
e
[
i
]
[
j
]
f[i][j]=min(f[i-1][j],f[i-1][j-1])+triangle [i][j]
f[i][j]=min(f[i−1][j],f[i−1][j−1])+triangle[i][j]
第0列只能从0列得到,固有:
f
[
i
]
[
0
]
=
f
[
i
−
1
]
[
0
]
+
t
r
i
a
n
g
l
e
[
i
]
[
0
]
f[i][0]=f[i-1][0]+triangle [i][0]
f[i][0]=f[i−1][0]+triangle[i][0]
第i列只能从i-1列得到,固有:
f
[
i
]
[
i
]
=
f
[
i
−
1
]
[
i
−
1
]
+
t
r
i
a
n
g
l
e
[
i
]
[
i
]
f[i][i]=f[i-1][i-1]+triangle [i][i]
f[i][i]=f[i−1][i−1]+triangle[i][i]
得出从上到下每行每个数字的最小路径,就能得到到达最后一行每个数字的最小路径,然后再选取最小值,就可以的到整体的最小路径。
class Solution {
public:
int minimumTotal(vector<vector<int>>& triangle) {
int n=triangle.size();
vector<vector<int>> f(n,vector<int>(n));
f[0][0]=triangle[0][0];
for(int i=1;i<n;++i){
f[i][0]=f[i-1][0]+triangle[i][0];
for(int j=1;j<i;j++){
f[i][j]=min(f[i-1][j-1],f[i-1][j])+triangle[i][j];
}
f[i][i]=f[i-1][i-1]+triangle[i][i];
}
return *min_element(f[n - 1].begin(), f[n - 1].end());
}
};
空间优化
因为i行只与i-1行有关,故只需要2n存储空间即可,即O(n);
class Solution {
public:
int minimumTotal(vector<vector<int>>& triangle) {
int n=triangle.size();
//vector<vector<int>> f(n,vector<int>(n));
vector<int> f(n,0);
vector<int> m(n,0);
f[0]=triangle[0][0];
for(int i=1;i<n;++i){
m[0]=f[0]+triangle[i][0];
for(int j=1;j<i;j++){
m[j]=min(f[j-1],f[j])+triangle[i][j];
}
m[i]=f[i-1]+triangle[i][i];
f=m;
//f.assign(m.begin(),m.end());
}
return *min_element(f.begin(), f.end());
}
};