问题描述:给定一个由n行数字组成的数字三角形,如下图所示。试用动态规划算法,计算出从三角
顶部至底部的一条路径,使得该路径经过的数字总和最大。
注意每个数字只能走向下一行左边或右边的数字,而不能跳跃的走。
7
3 8
8 1 0
2 7 4 4
4 5 2 6 5
输入格式
第一行是数字三角的行数n,1<=n<=100。接下来n行是数字三角各行中的数字,所有数字在0~99之间。
5
7
3 8
8 1 6
2 7 4 4
4 5 2 4 5
输出格式
输出两行,第一行是计算出的最大路径的和值,第二行是该路径上的数字。若有多条路径,靠右的路径
优先(即仅仅输出靠右的路径即可,无需多条路径都输出)。
30
7 8 6 4 5
分析:
输入为一个二维数组a,求的是以a[0][0]为顶部的最大路径和。以a[0][0]为顶部的最大路径和,等效于a[0][0]加上max{ 以a[0+1][0]为顶部的最大路径和,
以a[0+1][0+1]为顶部的最大路径和},一直往下划分。于是得到公式:P(i,j) = max{ P(i+1,j), P(i+1, j+1) } + a[i][j], 当 0 <= i < n-1;
P(i,j) = a[i][j], 当 i=n-1(最底层); 由于是个动态规划,我们用一个与a同维的数组b保存计算过的P(i,j)的值,这样就不会重复计算。
已经得到最大路径和,那么如何显示这一条路径?
这就要用到一个与a同维的数组s了,s[i][j]记录a[i][j]下一步是往左下走,还是右下走。往左下走,s[i][j] = 0; 右下走,s[i][j] = 1;
什么时候进行s[i][j]赋值呢? 当然是在判断左下大,还是右下大的时候。即求 max{ P(i+1,j), P(i+1, j+1) }时。
路径记录全部储存在s中后,如何输出呢?
void put(int i, int j){//输出以a[i][j]为顶部的路径
if (i == n) return;
cout << a[i][j] << " ";
put(i + 1, j + s[i][j]);
}
下面贴代码:
#include <iostream>
#include <cstdio>
#include <cmath>
#include <cstring>
using namespace std;
#define N 100
int n = 0;
int a[N][N], d[N][N], s[N][N];//数组a记录输入,d[i][j]记录以a[i][j]为顶部的最大路径和,s[i][j]记录向左下或者向右下
int longestPath(int a[N][N], int i, int j){//以a[i][j]为顶部的最大路径和
if(i==n-1) return d[i][j] = a[i][j];//最底层
if(d[i][j]) return d[i][j]; //已经算过就直接返回
int max = 0;
int left = longestPath(a,i+1,j);
int right = longestPath(a,i+1,j+1);
if(left > right){
max = left;
}else max = right, s[i][j] = 1;
return d[i][j] = (max + a[i][j]);
}
void put(int i, int j){//输出以a[i][j]为顶部的路径
if (i == n) return;
cout << a[i][j] << " ";
put(i + 1, j + s[i][j]);
}
int main()
{
cin >> n;
memset(a,0,sizeof(a));
memset(d,0,sizeof(d));
memset(s,0,sizeof(s));
for(int i=0; i<n; i++){
for(int j=0; j<=i; j++){
cin >> a[i][j];
}
}
cout<< longestPath(a,0,0) << endl;//以a[0][0]为顶部的最大路径和
put(0, 0);//输出最大路径
return 0;
}