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;
 }