E - Hatsune Miku （dp）

#include<bits/stdc++.h>
using namespace std;
const int M=55;
const int N=105;
int mp[M][M];
int f[N][M];
int a[N];
int main(){
	//freopen("in.txt","r",stdin);
	int T;scanf("%d",&T);
	while(T--){
		int n,m;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(f,0,sizeof(f));
		for(int i=2;i<=n;i++){
			if(a[i-1]!=-1){
				for(int j=1;j<=m;j++)
					f[i][j]=max(f[i][j],f[i-1][a[i-1]]+mp[a[i-1]][j]);
			}
			else{
				for(int j=1;j<=m;j++)
					for(int k=1;k<=m;k++)
						f[i][k]=max(f[i-1][j]+mp[j][k],f[i][k]); 
			}
		}
		int ans=0;
		for(int i=1;i<=m;i++){
			ans=max(f[n][i],ans);
		}
		printf("%d\n",ans);
	} 
	return 0;
}