Time limit
1000 ms
Memory limit
32768 kB
在讲述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代码:
#include<iostream>
#include<algorithm>
#include<string.h>
using namespace std;
const int maxn = 1e3+10;
int d[maxn][maxn],a[maxn][maxn];
int i,j,n,t;
int main()
{
ios::sync_with_stdio(false);
cin>>t;
while(t--)
{
cin>>n;
for(i=1;i<=n;i++)
{
for(j=1;j<=i;j++)
{
cin>>a[i][j];
}
}
for(int i=1;i<=n;i++) //将最后一行进行赋值,也就是将最后一行得值给d数组
d[n][i] = a[n][i];
for(i=n-1;i>=1;i--) //控制外层循环层数
{
for(j=1;j<=i;j++) //对应每行有多少个数
{
d[i][j] = max(d[i+1][j],d[i+1][j+1]) + a[i][j]; //取数的标准
}
}
cout<<d[1][1]<<endl; //输出d[1][1]就是最终的答案
}
return 0;
}
对于上述代码有空间上的优化:
#include<iostream>
#include<algorithm>
#include<string.h>
using namespace std;
const int maxn = 1e3+10;
int a[maxn][maxn];
int *maxsum;
int i,j,n,t;
int main()
{
ios::sync_with_stdio(false);
cin>>t;
while(t--)
{
cin>>n;
for(i=1;i<=n;i++)
{
for(j=1;j<=i;j++)
{
cin>>a[i][j];
}
}
maxsum = a[n]; //maxsum指向第n行
for(i=n-1;i>=1;i--) //控制外层循环层数
{
for(j=1;j<=i;j++) //对应每行有多少个数
{
maxsum[j] = max(maxsum[j],maxsum[j+1]) + a[i][j]; //取数的标准
}
}
cout<<maxsum[1]<<endl; //输出第一行的答案
}
return 0;
}
记忆化递归代码(不套个记忆化铁定超时,因为有大量已经不需要重新计算的子状态):
#include<iostream>
#include<algorithm>
#include<string.h>
using namespace std;
const int maxn = 100;
int n,maxsum[maxn][maxn];
int d[maxn][maxn];
int Maxsum(int i,int j)
{
if(maxsum[i][j]!=-1)
return maxsum[i][j];
if(i==n)
maxsum[i][j]=d[i][j];
else
{
int x = Maxsum(i+1,j);
int y = Maxsum(i+1,j+1);
maxsum[i][j] = max(x,y) + d[i][j];
}
return maxsum[i][j];
}
int main()
{
ios::sync_with_stdio(false);
int i,j,T;
cin>>T;
while(T--){
cin>>n;
for(i=1;i<=n;i++)
{
for(j=1;j<=i;j++)
{
cin>>d[i][j];
maxsum[i][j] = -1;
}
}
cout<<Maxsum(1,1)<<endl;}
return 0;
}