#数塔
在讲述DP算法的时候,一个经典的例子就是数塔问题,它是这样描述的:
有如下所示的数塔,要求从顶层走到底层,若每一步只能走到正下方或者右下方的节点,则经过的结点的数字之和最大是多少?
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
如果我们用纯dfs计算每一种走法的总和去取最大值,经过我亲测,TLE,所以我们需要想想办法。
瞎分析一波。。我们走每一种方法的时候,下面的点被重复走很多遍,所以我们需要把它存起来,下次找到这个点的时候直接用就可以了,而不需要重复计算。
#include<iostream>
#include<queue>
#include<string.h>
#include<string>
#include<stdio.h>
#include<math.h>
#include<algorithm>
using namespace std;
typedef long long ll;
const int N=1e3+5;
int mp[N][N],dp[N][N];
int n,sum=0;
int dfs(int x,int y)
{
if(dp[x][y])
return dp[x][y];//如果这个点被搜索过,直接返回值
if(x==n)
return mp[x][y];
dp[x][y]=max(mp[x][y]+dfs(x+1,y),mp[x][y]+dfs(x+1,y+1));//向下寻找
return dp[x][y];
}
int main()
{
std::ios::sync_with_stdio(false);
cin.tie(0);
cout.tie(0);
int t;
cin>>t;
while(t--)
{
cin>>n;
for(int i=1;i<=n;i++)
{
for(int j=1;j<=i;j++)
{
cin>>mp[i][j];
}
}
cout<<dfs(1,1)<<endl;
memset(dp,0,sizeof dp);
memset(mp,0,sizeof mp);
}
}