4.6 数塔

4.6数塔

【问题描述】

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

9

12

15

10

6

8

2

18

9

5

19

7

10

4

16

【输入形式】

输入时第一行一个整数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

动态规划法

思路：
矩阵数组dp存储结果，将每个到达叶结点的路径最大值都求出来；
在每一步中，状态转移方程为：
dp[i][j] = max(dp[i-1][j-1],dp[i-1][j])+tower[i][j];
（其中两边的边界情况应该单独考虑，即代码中的if语句）
即每次动规，都要求出，到达该结点的路径的最大值。
寻找路径：
向上找路径的时候，每次选择最大值，沿路径dp数组向上

//6、数塔
#include <iostream>
#include <algorithm>
#include <stack>
using namespace std;
int main(){
	int n;
	cin >> n;
	int **tower;
	tower = new int*[n];
	for(int i=0;i<n;i++)
		tower[i] = new int[i+1];
	for(int i=0;i<n;i++)
		for(int j=0;j<i+1;j++)
			cin >> tower[i][j];
	int dp[n][n];
	dp[0][0] = tower[0][0];//结果储存
	for(int i=1;i<n;i++)
		for(int j=0;j<i+1;j++){//状态转移方程
			if(j == 0)//边界情况
				dp[i][j] = dp[i-1][j]+tower[i][j]; 
			else if(j == i)//另一边边界
				dp[i][j] = dp[i-1][j-1]+tower[i][j];
			else//普通中间情况
				dp[i][j] = max(dp[i-1][j-1],dp[i-1][j])+tower[i][j];	
		}
	int m=0,index[n];//向上寻找路径，反向操作即可
	for(int i=0;i<n;i++)
		if(m < dp[n-1][i]){
			m = dp[n-1][i];
			index[n-1] = i;
		}
	cout << m << endl;
	for(int i=n-2;i>=0;i--){
		if(index[i+1] == 0)
			index[i] = 0;
		else if(index[i+1] == i+1)
			index[i] = i;
		else
			index[i] = (dp[i][index[i+1]-1]>dp[i][index[i+1]]?index[i+1]-1:index[i+1]); 
	} 
	for(int i=0,j=0;i<n;i++,j++)
		cout << tower[i][index[j]]<<" ";
	cout << endl;
}