HDU 2084
一、题目
在讲述DP算法的时候,一个经典的例子就是数塔问题,它是这样描述的:有如下所示的数塔,要求从顶层走到底层,若每一步只能走到相邻的结点,则经过的结点的数字之和最大是多少?
输入
输入数据首先包括一个整数C,表示测试实例的个数,每个测试实例的第一行是一个整数N(1 <= N <= 100),表示数塔的高度,接下来用N行数字表示数塔,其中第i行有个i个整数,且所有的整数均在区间[0,99]内。
输出
对于每个测试实例,输出可能得到的最大和,每个实例的输出占一行。
二、思路
该题有两种解题思路:
1. 顺推法
从数塔的顶部向下推,每个数(除去边界)向上有两个分支,比较上面两个分支数(指的是加完后的新的数塔)的大小,哪个数大就将那个数再加到本位置上面,直到最后一层。则最大的数字之和就是最后一行中最大的那个数。
其动态规划方程为:
F[0][0] = A[0][0];
F[i][0] = F[i-1][0]+A[i][0];
F[i][i] = F[i-1][i-1]+A[i][i];
F[i][j] = max{F[i-1][j-1],F[i-1][j]}+A[i][j];
最终答案为max{F[n-1][j],0<=j<=n-1};
2. 逆推法
从数塔的最低一层开始向上推,比较相邻的两个数(加完后的数)的大小,哪个数大就加到上一层中的那个数上面,直到加到最上面一层,则最大的数就是第一层的那个数。
其动态规划方程为:
G[n-1][j] = A[n-1][j] 0<=j<=n-1;
G[i][j] = max{G[i+1][j],G[i+1][j+1]}+A[i][j];
最后答案为G[0][0];
三、代码实现
1.顺推法
#include<stdio.h>
#define N 100
int A[N][N];
int F[N][N];
int main()
{
int i,j,n;
scanf("%d",&n);
for(i = 0; i < n; i++)
for(j = 0; j <= i; j++) //注意j是从0到i,而不是从0到n-1
scanf("%d",&A[i][j]); //将数塔存入A[N][N]数组中。
F[0][0] = A[0][0]; //边界处理
for(i = 1; i < n; i++)
for(j = 0; j < n; j++)
{
if(j == 0)
F[i][j] = A[i][j] + F[i - 1][j]; //边界处理
else if(j == i)
F[i][j] = A[i][j] + F[i - 1][j - 1]; //边界处理
else
{
if(F[i - 1][j - 1] > F[i - 1][j]) //动态规划方程核心
F[i][j] = F[i - 1][j - 1] + A[i][j];
else
F[i][j] = F[i - 1][j] + A[i][j];
}
}
int max = F[n - 1][0];
for(j = 1; j < n; j++) //求最大值MAX
if(max < F[n - 1][j]) max = F[n - 1][j];
printf("%d\n",max);
return 0;
}
2.逆推法
#include<stdio.h>
#define N 100
int A[N][N];
int G[N][N];
int main()
{
int i,j,n;
scanf("%d",&n);
for(i = 0; i < n; i++)
for(j = 0; j <= i; j++) //注意j是从0到i,而不是从0到n-1
scanf("%d",&A[i][j]); //将数塔存入A[N][N]数组中。
for(j = 0; j < n; j++) //边界处理
G[n - 1][j] = A[n - 1][j];
for(i = n - 2; i >= 0; i--) //注意i的范围
for(j = 0; j <= i; j++)
{
if(G[i + 1][j] > G[i + 1][j + 1])
G[i][j] = G[i + 1][j] + A[i][j];
else
G[i][j] = G[i + 1][j + 1] + A[i][j];
}
printf("%d",G[0][0]); //结果就是G[0][0]
return 0;
}