003-2算法笔记【动态规划】习题3-4-数字三角形问题

数字三角形问题

题目

给定一个由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;
    }
};