Problem Description
在讲述DP算法的时候,一个经典的例子就是数塔问题,它是这样描述的:
有如下所示的数塔,要求从顶层走到底层,若每一步只能走到相邻的结点,则经过的结点的数字之和最大是多少?
已经告诉你了,这是个DP的题目,你能AC吗?
Input
输入数据首先包括一个整数C,表示测试实例的个数,每个测试实例的第一行是一个整数N(1 <= N <= 100),表示数塔的高度,接下来用N行数字表示数塔,其中第i行有个i个整数,且所有的整数均在区间[0,99]内。
Output
对于每个测试实例,输出可能得到的最大和,每个实例的输出占一行。
Sample Input
1
5
7
3 8
8 1 0
2 7 4 4
4 5 2 6 5
Sample Output
30
递推方程:
dp[0][0]=a[0][0]; i=0,j=0;
i!=0时:
dp[i][0]=a[i][0]+dp[i-1][0]; j=0;
dp[i][j]=a[i][j]+a[i-1][j-1]; j=i;
dp[i][j]=a[i][j]+max(dp[i-1][j-1]+dp[i-1][j]); 除了以上两种情况
代码如下:
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <iostream>
using namespace std;
const int maxn=105;
int c,n;
int a[maxn][maxn];
int dp[maxn][maxn];
int Max;
void init()
{
memset (dp,0,sizeof(dp));
Max=0;
}
int main()
{
scanf("%d",&c);
while (c--)
{
init();
scanf("%d",&n);
for (int i=0;i<n;i++)
for (int j=0;j<=i;j++)
scanf("%d",&a[i][j]);
Max=dp[0][0]=a[0][0];
for (int i=1;i<n;i++)
{
for (int j=0;j<=i;j++)
{
if(j==0)
dp[i][j]=a[i][j]+dp[i-1][j];
else if(j==i)
dp[i][j]=a[i][j]+dp[i-1][j-1];
else
dp[i][j]=a[i][j]+max(dp[i-1][j-1],dp[i-1][j]);
Max=max(Max,dp[i][j]);
}
}
printf("%d\n",Max);
}
return 0;
}