dp[i][j]表示处理到第i位时最后一个数是j的最大价值。
转移就是dp[i][j]=max(dp[i-1][k]+cost[k][j]) (枚举k)
如果遇到给定的数,就把处理到这一位不是这个给定的数的情况全部处理为负无穷,因为是不合法的。初始化时如果第一个数就给定了也是这样处理。
代码:
#include <cstdio>
#include <cmath>
#include <cstring>
#include <iostream>
#include <algorithm>
#include <cmath>
#define INF (-1000000)
#define LL long long
using namespace std;
int dp[105][55];
int cost[55][55];
int p[105];
int n,m;
int main()
{
int t;
cin>>t;
while(t--)
{
scanf("%d%d",&n,&m);
for(int i=1;i<=m;i++){
for(int j=1;j<=m;j++){
scanf("%d",&cost[i][j]);
}
}
for(int i=1;i<=n;i++)
scanf("%d",&p[i]);
memset(dp,0,sizeof(dp));
if(p[1]!=-1){
for(int i=1;i<=m;i++){
dp[1][i]=INF;
}
dp[1][p[1]]=0;
}
else {
for(int i=1;i<=m;i++){
dp[1][m]=0;
}
}
for(int i=2;i<=n;i++){
if(p[i]!=-1){
for(int j=1;j<=m;j++){
dp[i][p[i]]=max(dp[i-1][j]+cost[j][p[i]],dp[i][p[i]]);
}
for(int j=1;j<=m;j++){
if(j!=p[i]) dp[i][j]=INF;
}
}
else{
for(int j=1;j<=m;j++){
for(int k=1;k<=m;k++){
dp[i][k]=max(dp[i-1][j]+cost[j][k],dp[i][k]);
}
}
}
}
int res=0;
for(int i=1;i<=m;i++){
res=max(res,dp[n][i]);
}
cout<<res<<endl;
}
return 0;
}