每日一题——7-1 数字三角形问题 (15 分)

7-1 数字三角形问题 (15 分)

给定一个由n行数字组成的数字三角形如下图所示。试设计一个算法，计算出从三角形的顶至底的一条路径，使该路径经过的数字总和最大。

对于给定的由n行数字组成的数字三角形，计算从三角形的顶至底的路径经过的数字和的最大值。

输入格式:

输入数据的第1行是数字三角形的行数n，1≤n≤100。接下来n行是数字三角形各行中的数字。所有数字在0…99之间。

输出格式:

输出数据只有一个整数，表示计算出的最大值。

输入样例:

在这里给出一组输入。例如：

5
7
3 8
8 1 0
2 7 4 4
4 5 2 6 5

输出样例:

在这里给出相应的输出。例如：

30

思路

一个相对简单的dp难问题，题目中没有详说是否每行只能取一个，但是我后面看了下示例，应该是只能由左上方与右上方取到；所以表达式就很明显了

$d{p}_{i,j}=\{\begin{array}{l}d{p}_{i-1,0}+matri{x}_{i,0}j=0\\ d{p}_{i-1,j-1}+matri{x}_{i,j}j=i\\ max(d{p}_{i-1,j},d{p}_{i-1,j-1})+matrix[i][j]others\end{array}$

for(i=1;i<n;i++){
        for(j=0;j<=i;j++){
            if(j==0){
                dp[i][j]=dp[i-1][j]+matrix[i][j];
            }
            else if (j==i){
                dp[i][j]=dp[i-1][j-1]+matrix[i][j];
            }
            else{
                dp[i][j]=max(dp[i-1][j],dp[i-1][j-1])+matrix[i][j];
            }
        }
    }

代码

#include <bits/stdc++.h>
using namespace std;

int main(){
    short n,i,j;
    int max_ele=0;
    int dp[101][101];
    short matrix[101][101];
    cin>>n;
    for(i=0;i<n;i++){
        for(j=0;j<=i;j++){
            cin>>matrix[i][j];
        }
    }
    //dp
    dp[0][0]=matrix[0][0];
    for(i=1;i<n;i++){
        for(j=0;j<=i;j++){
            if(j==0){
                dp[i][j]=dp[i-1][j]+matrix[i][j];
            }
            else if (j==i){
                dp[i][j]=dp[i-1][j-1]+matrix[i][j];
            }
            else{
                dp[i][j]=max(dp[i-1][j],dp[i-1][j-1])+matrix[i][j];
            }
        }
    }
    for(i=0;i<n;i++){
        if(dp[n-1][i]>max_ele){
            max_ele=dp[n-1][i];
        }
    }
    cout<<max_ele<<endl;
    return 0;
}