hdu 5074

http://acm.hdu.edu.cn/showproblem.php?pid=5074

#include<stdio.h>
#include<stdlib.h>
#include<math.h>
#include<string.h>
#include<iostream>
#include<algorithm>
using namespace std;


int dp[110][60];
int Map[60][60];
int a[110];


int main()
{
    int m,n;
    int T;
    int i,j;
    int mi,mj;
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d%d",&n,&m);
        for(i=1;i<=m;i++)
            for(j=1;j<=m;j++)
                scanf("%d",&Map[i][j]);
        for(i=1;i<=n;i++)
            scanf("%d",&a[i]);
        memset(dp,0,sizeof(dp));
        for(i=2;i<=n;i++)
        {
            if(a[i-1]>0&&a[i]>0)
                dp[i][a[i]]=dp[i-1][a[i-1]]+Map[a[i-1]][a[i]];


            else
            if(a[i-1]>0&&a[i]<0)
               for(j=1;j<=m;j++)
                    dp[i][j]=dp[i-1][a[i-1]]+Map[a[i-1]][j];


            else
            if(a[i-1]<0&&a[i]<0)
                for(mi=1;mi<=m;mi++)
                     for(mj=1;mj<=m;mj++)
                        dp[i][mi]=max(dp[i][mi],dp[i-1][mj]+Map[mj][mi]);


            else
                for(mi=1;mi<=m;mi++)
                    dp[i][a[i]]=max(dp[i][a[i]],dp[i-1][mi]+Map[mi][a[i]]);
        }
        int Max=0;
        if(a[n]<0)
            for(i=1;i<=m;i++)
                Max=max(Max,dp[n][i]);
        else
            Max=dp[n][a[n]];
        printf("%d\n",Max);
    }
    return 0;
}