dp递推出第n个音符以1-m种音符结尾的最大值,然后输出这m个数的最大值,239ms,看前辈大牛14ms过的,不明觉厉。。
#include<stdio.h>
#include<string.h>
#include<cmath>
#include<cstdlib>
#include<set>
#include<queue>
using namespace std;
int n,m,i,j,k,t,x;
int a[200]={0},b[200]={0},ap[100][100],s[50];
int main()
{
scanf("%d",&t);
while(t--)
{
memset(a,0,sizeof(a));
scanf("%d%d",&n,&m);
for(i=1;i<=m;i++)
for(j=1;j<=m;j++)scanf("%d",&ap[i][j]);
for(i=0;i<n;i++)scanf("%d",&s[i]);
for(i=1;i<n;i++)
{
if(s[i]==-1&&s[i-1]==-1){
for(j=1;j<=m;j++)b[j]=a[j];
for(j=1;j<=m;j++)
{
int e=b[j];
for(k=1;k<=m;k++)if(a[k]<e+ap[j][k])a[k]=e+ap[j][k];
}
}
else if(s[i]==-1)
{
int e=a[s[i-1]];
for(j=1;j<=m;j++)a[j]=e+ap[s[i-1]][j];
}
else if(s[i-1]==-1)
{
for(j=1;j<=m;j++)b[j]=a[j];
int big=b[1]+ap[1][s[i]];
for(j=2;j<=m;j++)
{
if(big<b[j]+ap[j][s[i]])big=b[j]+ap[j][s[i]];
}
a[s[i]]=big;
}
else a[s[i]]=a[s[i-1]]+ap[s[i-1]][s[i]];
}
x=a[1];
for(i=2;i<=m;i++)
if(a[i]>x)x=a[i];
printf("%d ",x);
memset(a,0,sizeof(a));
for(i=1;i<n;i++)
{
for(j=1;j<=m;j++)b[j]=a[j];
for(j=1;j<=m;j++)
{
int e=b[j];
for(k=1;k<=m;k++)if(a[k]<e+ap[j][k])a[k]=e+ap[j][k];
}
}
x=a[1];
for(i=2;i<=m;i++)
if(a[i]>x)x=a[i];
printf("%d\n",x);
}
return 0;
}