ACM/ICPC2014鞍山現場賽E hdu5074Hatsune Miku

題目連結：http://acm.hdu.edu.cn/showproblem.php?pid=5074

題意：

給定一個m*m的矩陣mp，然後給定一個長度為n的序列

sum= mp[a[1]][a[2]]+......+mp[a[n-1]][a[n]];

如果a[i]小於0的話則a[i]可以為1~m之間的任意一個數，求sum的最大值

我們可以列舉第i個位置

然後dp[i][j]=max(dp[i][j],dp[i-1][k]+mp[k][j]) 表示第i個位置在j的最大值；

然後分兩種情況

如果a[i]是一個確定的數的話 dp[i][a[i]]=max( dp[i][a[i]],dp[i-1][j]+mp[j][a[i]] )

如果不確定的話 就只能列舉a[i]了 dp[i][k]=max(dp[i][k],dp[i-1][j]+mp[j][k]);

這兩種情況都是建立在前一個位置已經確定了值的情況下；

程式碼如下：

#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;

const int maxn = 55;
int mp[maxn][maxn];
int dp[maxn*2][maxn],a[maxn*2];

int main()
{
    int t,n,m;
    scanf("%d",&t);
    while(t--){
        scanf("%d%d",&n,&m);
        for(int i=1;i<=m;i++){
            for(int j=1;j<=m;j++)
                scanf("%d",&mp[i][j]);
        }
        for(int i=1;i<=n;i++)
            scanf("%d",&a[i]);
        memset(dp,-1,sizeof(dp));
        if(a[1]!=-1)
            dp[1][a[1]]=0;
        else
            for(int i=1;i<=m;i++)
                dp[1][i]=0;
        for(int i=2;i<=n;i++){
            for(int j=1;j<=m;j++){
                if(dp[i-1][j]!=-1){//如果前一項已經確定
                    if(a[i]!=-1)
                        dp[i][a[i]]=max(dp[i][a[i]],dp[i-1][j]+mp[j][a[i]]);
                    else{
                        for(int k=1;k<=m;k++)
                            dp[i][k]=max(dp[i][k],dp[i-1][j]+mp[j][k]);
                    }
                }
            }
        }
        /***************
        cout<<"   debug   "<<endl;
        for(int i=1;i<=n;i++){
            for(int j=0;j<=m;j++)
                cout<<dp[i][j]<<" ";
            cout<<endl;
        }
        cout<<"***********"<<endl;
        **************/
        int maxn = -1000;
        for(int i=1;i<=m;i++)
            maxn=max(dp[n][i],maxn);
        printf("%d\n",maxn);
    }
    return 0;
}