又见数塔
题目描述
在讲述DP算法的时候,一个经典的例子就是数塔问题,它是这样描述的:
有如下所示的数塔,要求从顶层走到底层,若每一步只能走到相邻的结点,则经过的结点的数字之和最大是多少?并输出得到最大值的路径。
输入
输入数据首先包括一个整数C,表示测试实例的个数,每个测试实例的第一行是一个整数N(1 <= N <= 100),表示数塔的高度,接下来用N行数字表示数塔,其中第i行有个i个整数,且所有的整数均在区间[0,99]内。
输出
对于每个测试实例,输出可能得到的最大和,并输出其路径。如果有多条路径则输出最左边的那条路径。
样例输入
1
5
7
3 8
8 1 0
2 7 4 4
4 5 2 6 5
样例输出
30
(0,0)-->(1,0)-->(2,0)-->(3,1)-->(4,1)
#include<iostream>#include<stdio.h>#include<string.h>using namespace std;int main(){int a[102][102],b[102][102],num,n,i,j,mmax;scanf("%d",&num);while(num--){scanf("%d",&n);memset(a,0,sizeof(a));memset(b,0,sizeof(b));for(i=0;i<n;i++)for(j=0;j<=i;j++){scanf("%d",&a[i][j]);b[i][j]=a[i][j];}for(i=n-2;i>=0;i--){for(j=0;j<=i;j++){
if(a[i+1][j]>a[i+1][j+1])a[i][j]+=a[i+1][j];elsea[i][j]+=a[i+1][j+1];}}mmax=a[0][0];printf("%d\n",mmax);for(i=0;i<n;i++)for(j=0;j<=i;j++){if(mmax==a[i][j]){if(i==n-1)printf("(%d,%d)\n",i,j);elseprintf("(%d,%d)-->",i,j);mmax=mmax-b[i][j];}}}return 0;}
解题思路:先定义两个二维数组,一个是a,一个是b,a中用于计算,b中保留原始数据,在a,中从n-2行往上走(即倒数第二行),每个数都加上它(下面的数和斜下角的数中的最大值),最后在塔顶得到的就是最大值,然后在从塔顶开始走,减去b中的原始数据,开始与第二行的一个一个比较,遇到相等的,把坐标输出来,然后在减去b中这个位置的数,继续与a中下一行进行比较,找到相等的,就将坐标输出,当i==n-1,说明找到了最后一个,那么只用输出这个坐标即可,不需要加箭头。例:0 1 2 3 40 91 12 152 10 6 83 2 18 9 54 19 7 10 4 16
逆着走//从倒数第二行开始,每个数都加上它(下面的数和斜下角的数的最大值//59 (0.0) 59-9=5050 49 (1,0) 50-12=3838 34 29 (2,0) 38-10=2821 28 19 21 (3,1) 28-18=1019 7 10 4 16 (4,2)