1. 程式人生 > >poj1163 數字三角形 (動態規劃)

poj1163 數字三角形 (動態規劃)

試題連結: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;
}