数塔
Problem Description
在讲述DP算法的时候,一个经典的例子就是数塔问题,它是这样描述的:
有如下所示的数塔,要求从顶层走到底层,若每一步只能走到相邻的结点,则经过的结点的数字之和最大是多少?
已经告诉你了,这是个DP的题目,你能AC吗?
有如下所示的数塔,要求从顶层走到底层,若每一步只能走到相邻的结点,则经过的结点的数字之和最大是多少?
已经告诉你了,这是个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
相当经典的一题,下面我给普通方法和记忆化搜索两种方法。
记忆化搜索:
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
#define maxn 105
int map[maxn][maxn];
int dp[maxn][maxn];
int vis[maxn][maxn];
int n,t;
int fun(int x,int y);
int main(){
scanf("%d",&t);
while(t--){
cin >>n;
for(int i = 1;i <= n; i++){
for(int j = 1; j <= i; j++){
scanf("%d",&map[i][j]);
}
}
memset(dp,0,sizeof(dp));
memset(vis,-1,sizeof(vis));
int ans = fun(1,1);
printf("%d\n",ans);
}
return 0;
}
int fun(int x, int y){
if(x == n || y == n){
return map[x][y];
}
if(vis[x][y] != -1){
return vis[x][y];
}
return vis[x][y] = max(fun(x+1,y+1),fun(x+1,y)) + map[x][y];
}
普通递推方程:
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
#define maxn 105
int map[maxn][maxn];
int dp[maxn][maxn];
int n,t;
int main(){
scanf("%d",&t);
while(t--){
scanf("%d",&n);
for(int i = 1; i <= n; i++){
for(int j = 1; j <= i; j++){
scanf("%d",&map[i][j]);
}
}
memset(dp,0,sizeof(dp));
for(int i = n; i>0;i--){
for(int j = 1; j <= i; j++){
dp[i][j] = max(dp[i+1][j],dp[i+1][j+1]) +map[i][j];
}
}
printf("%d\n",dp[1][1]);
}
return 0;
}