动态规划(一)
例题一、数字三角形(POJ1163)
在上面的数字三角形中寻找一条从顶部到底边的路径,使得路径上所经过的数字之和最大。路径上的每一步都只能往左下或右下走。只需要求出这个最大和即可,不必给出具体路径。
三角形的行数大于1小于等于100,数字为 0 - 99
输入格式:
5 //三角形行数。下面是三角形
7
3 8
8 1 0
2 7 4 4
4 5 2 6 5
要求输出最大和
解题思路:
用二维数组存放数字三角形。D( r, j) : 第r行第 j 个数字(r,j从1开始算)MaxSum(r, j) : 从D(r,j)到底边的各条路径中,最佳路径的数字之和。
问题:求 MaxSum(1,1)
典型的递归问题。
D(r, j)出发,下一步只能走D(r+1,j)或者D(r+1, j+1)。故对于N行的三角形:
if ( r == N)
MaxSum(r,j) = D(r,j)
else
MaxSum( r, j) = Max{ MaxSum(r +1,j), MaxSum(r+1,j+1) }+ D(r,j)
下面是简单的递归,但存存大量的重复计算。
#include<iostream>
using namespace std;
#define Max 101
int D[Max][Max];
int n;
int MaxSum(int i,int j){
if(i==n-1) return D[i][j];
int y=MaxSum(i+1,j);
int z=MaxSum(i+1,j+1);
if(z>y)y=z;
return y+D[i][j];
}
int main(){
cin>>n;
for(int i=0;i<n;++i)
for(int j=0;j<i+1;++j)
cin>>D[i][j];
for(int i=0;i<n;++i){
for(int j=0;j<i+1;++j)
cout<<D[i][j]<<" ";
cout<<endl;
}
cout<<MaxSum(0,0)<<endl;
return 0;
}
为什么超时?
回答:重复计算
如果采用递归的方法,深度遍历每条路径,存在大量重复计算。则时间复杂度为 2 n ,对于 n = 100行,肯定超时。
改进:如果每算出一个MaxSum(r,j)就保存起来,下次用到其值的时候直接取用,则可免去重复计算。那么可以用O(n 2 )时间完成计算。因为三角形的数字总数是 n(n+1)/2
数字三角形的记忆递归型动归程序:
#include<iostream>
using namespace std;
#define Max 101
int D[Max