在讲述DP算法的时候,一个经典的例子就是数塔问题,它是这样描述的:
有如下所示的数塔,要求从顶层走到底层,若每一步只能走到相邻的结点,则经过的结点的数字之和最大是多少?
已经告诉你了,这是个DP的题目,你能AC吗?
Input
输入数据首先包括一个整数C,表示测试实例的个数,每个测试实例的第一行是一个整数N(1 <= N <= 100),表示数塔的高度,接下来用N行数字表示数塔,其中第i行有个i个整数,且所有的整数均在区间[0,99]内。
Output
对于每个测试实例,输出可能得到的最大和,每个实例的输出占一行。
1
5
7
3 8
8 1 0
2 7 4 4
4 5 2 6 5
Sample Output
30
从底层向上推,要使经过某一点dp[i][j]的和最大,那么就要求选择它下一层的dp[i+1][j]和dp[i+1][j+1]中大的一个,不难写出状态转移方程为dp[i][j] = max(dp[i+1][j], dp[i+1][j])+arr[i][j],注意最底层dp[n-1][j]应该初始为数塔最底层的值
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<cmath>
#define ll long long
#define rep(i,a,b)for(int i=a;i<=b;i++)
#define rep1(i,a,b)for(int i=a;i<b;i++)
#define per(i,a,b)for(int i=a;i>=b;i--)
using namespace std;
const int maxn=105;
int n,m;
int dp[maxn][maxn];
int sum[maxn][maxn];
int main(){
scanf("%d",&n);
while(n--){
scanf("%d",&m);
rep(i,1,m)
rep(j,1,i)
scanf("%d",&sum[i][j]);
rep(i,1,m){
dp[m][i]=sum[m][i];}
per(i,m-1,1)
rep(j,1,i)
dp[i][j]=max(dp[i+1][j],dp[i+1][j+1])+sum[i][j];
printf("%d\n",dp[1][1]);
}
return 0;
}