数字三角形问题
题目
给定一个由n行数字组成的数字三角形如下图所示。试设计一个算法,计算出从三角形的顶至底的一条路径,使该路径经过的数字总和最大。
5
7
3 8
8 1 0
2 7 4 4
4 5 2 6 5
输入
输入数据的第1行是数字三角形的行数n,接下来n行是数字三角形各行中的数字
输出
计算出的最大值
30
分析
f[i][j]=max(f[i-1][j-1],f[i-1][j])+triangle[i][j]
代码
#include<iostream>
#include<vector>
using namespace std;
int main(){
int n;
cin>>n;
vector<vector<int> > a(n+1,vector<int>(n+1));
for(int i=1;i<=n;i++)
for(int j=1;j<=i;j++)
cin>>a[i][j];
vector<vector<int> > f(n+1,vector<int>(n+1));
f[1][1]=a[1][1];
for(int i=2;i<=n;i++){
for(int j=1;j<=i;j++)
f[i][j]=max(f[i-1][j-1],f[i-1][j])+a[i][j];
}
int res=f[n][1];
for(int j=2;j<=n;j++)
res=max(res,f[n][j]);
cout<<res<<endl;
return 0;
}
leetcode: 剑指 Offer II 100. 三角形中最小路径之和
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],f[i-1][j-1])+triangle[i][j];
}
f[i][i]=f[i-1][i-1]+triangle[i][i];
}
int res=f[n-1][0];
for(int i=1;i<n;i++)
res=min(res,f[n-1][i]);
return res;
}
};