在讲述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[i][j]表示从点(i,j)出发,得到的最大和
int solve(int i,int j)
{
if(dp[i][j]>=0)return dp[i][j];//优化
if(i==n)return dp[i][j]=a[i][j];//最后一层
else return dp[i][j]=a[i][j]+max(solve(i+1,j),solve(i+1,j+1));//递归下一层
}
代码
#include<iostream>
#include<algorithm>
#include<cstring>
using namespace std;
int a[105][105];
int dp[105][105];
int n;
int solve(int i,int j)
{
if(dp[i][j]>=0)return dp[i][j];
if(i==n)return dp[i][j]=a[i][j];
else return dp[i][j]=a[i][j]+max(solve(i+1,j),solve(i+1,j+1));
}
int main()
{
int C;
cin>>C;
while(C--)
{
cin>>n;
for(int i=1;i<=n;i++)
for(int j=1;j<=i;j++)
cin>>a[i][j];
memset(dp,-1,sizeof(dp));
cout<<solve(1,1)<<endl;
}
return 0;
}
法二:递推
思路:从下而上推,dp[i][j]存储点(i,j)到最底层的最大和,
dp[i][j]=a[i][j]+max(dp[i+1][j],dp[i+1][j+1]),最后dp[1][1]便是答案
for(int j=1;j<=n;j++)
dp[n][j]=a[n][j];
for(int i=n-1;i>=1;i--)
for(int j=1;j<=i;j++)
{
dp[i][j]=a[i][j]+max(dp[i+1][j],dp[i+1][j+1]);
}
代码
#include<iostream>
#include<algorithm>
#include<cstring>
using namespace std;
int a[105][105];
int dp[105][105];
int n;
int main()
{
int C;
cin>>C;
while(C--)
{
cin>>n;
for(int i=1;i<=n;i++)
for(int j=1;j<=i;j++)
cin>>a[i][j];
for(int j=1;j<=n;j++)
dp[n][j]=a[n][j];
for(int i=n-1;i>=1;i--)
for(int j=1;j<=i;j++)
{
dp[i][j]=a[i][j]+max(dp[i+1][j],dp[i+1][j+1]);
}
cout<<dp[1][1]<<endl;
}
return 0;
}