poj1163 數字三角形 (動態規劃)
阿新 • • 發佈:2019-01-23
試題連結:http://poj.org/problem?id=1163
1.記憶遞迴型(自頂向下)
D[i][j]來存數字
典型的遞迴問題: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)用ans[i][j]來存計算出的結果,避免重複計算
#include <iostream> using namespace std; #define MAX 101 int D[MAX][MAX]; int ans[MAX][MAX]; int n; int MaxSum(int i,int j) { if(ans[i][j]!=-1) return ans[i][j]; if(i==n) ans[i][j]=D[i][j]; else ans[i][j]=max(MaxSum(i+1,j),MaxSum(i+1,j+1))+D[i][j]; return ans[i][j]; } int main() { cin>>n; for(int i=1; i<=n; i++) { for(int j=1; j<=i; j++) { cin>>D[i][j]; ans[i][j]=-1; } } cout<<MaxSum(1,1); return 0; }
2.迴圈遞推法(自底向上)
#include <iostream> using namespace std; #define MAX 101 int D[MAX][MAX]; int ans[MAX][MAX]; int n; int main() { cin>>n; for(int i=1; i<=n; i++) //讀入數字 for(int j=1; j<=i; j++) cin>>D[i][j]; for(int j=1; j<=n; j++) //先初始化最後一行 ans[n][j]=D[n][j]; for(int i=n-1; i>=1; i--) { for(int j=1; j<=n; j++) ans[i][j]=max(ans[i+1][j],ans[i+1][j+1])+D[i][j]; } cout<<ans[1][1]<<endl; return 0; }
空間優化
#include <iostream> using namespace std; #define MAX 101 int D[MAX][MAX]; int *ans; int n; int main() { cin>>n; for(int i=1; i<=n; i++) //讀入數字 for(int j=1; j<=i; j++) cin>>D[i][j]; ans=D[n]; //ans指向第n行 for(int i=n-1; i>=1; i--) for(int j=1; j<=i; j++) ans[j]=max(ans[j],ans[j+1])+D[i][j]; cout<<ans[1]<<endl; return 0; }