【程序设计训练】4-6 数塔

问题描述

给定一个数塔，如下图所示。在此数塔中，从顶部出发，在每一节点可以选择走左下或右下，一直走到底层。请找出一条路径，使路径上的数值和最大。

输入形式

输入时第一行一个整数n，表示该数塔的行数，其余n行表示该塔每行的数值

输出形式

输出包含两行，第一行为最大路径上的数值之和， 第二行n个数字为从上而下最大路径数值

样例输入

5
9
12 15
10 6 8
2 18 9 5
19 7 10 4 16

样例输出

59
9 12 10 18 10

参考代码

#include<iostream>
#include<cmath>
using namespace std;

int num[100][100];
int dp[100][100];
int re[50];
int main(){
	int n;cin>>n;

	for(int i=1;i<=n;i++){
		for(int j=1;j<=i;j++){
			cin>>num[i][j];
		}
	}
	for(int i=1;i<=n;i++){
		dp[n][i]=num[n][i];
	}
	for(int i=n-1;i>=1;i--){
		for(int j=1;j<=i;j++){
			dp[i][j]=max(dp[i+1][j],dp[i+1][j+1])+num[i][j];
		}
	} 
	int a=1,b=1; 
	for(int i=1;i<=n;i++){
		if(dp[a+1][b]>dp[a+1][b+1]){
			re[i]=dp[a][b]-dp[a+1][b];
			a++;
		}
		else {
			re[i]=dp[a][b]-dp[a+1][b+1];
			a++;b++;
		}
	}
	cout<<dp[1][1]<<endl;
	for(int i=1;i<=n;i++){
        cout<<re[i]<<" ";
    }  
	return 0;                                      
}