题意:要用一些片段组成一首歌,每两个片段的前后组合有一个好听度,歌曲的好听度是所有相邻片段好听度的和,一首歌由n个片段组成,有些片段可以换,有些不行,求歌曲最大的好听成都
思路:DP
代码如下:
#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
int dp[105][55];
int sav[55][55];
int A[105];
int main()
{
// freopen("data.txt","r",stdin);
int T;
scanf("%d",&T);
while(T--)
{
int n,m;
scanf("%d%d",&n,&m);
for(int i=0;i<m;++i)
{
for(int j=0;j<m;++j)
{
scanf("%d",&sav[i][j]);
}
}
for(int i=0;i<n;++i){scanf("%d",&A[i]);A[i]-=1;}
memset(dp,0,sizeof(dp));
for(int i=1;i<n;++i)
{
if(A[i]<0)
for(int j=0;j<m;++j)
{
if(A[i-1]<0)
for(int k=0;k<m;++k)
{
dp[i][j]=max(dp[i][j],dp[i-1][k]+sav[k][j]);
}
else
{
int k=A[i-1];
dp[i][j]=dp[i-1][k]+sav[k][j];
}
}
else
{
int j=A[i];
if(A[i-1]<0)
for(int k=0;k<m;++k)
{
dp[i][j]=max(dp[i][j],dp[i-1][k]+sav[k][j]);
}
else
{
int k=A[i-1];
dp[i][j]=dp[i-1][k]+sav[k][j];
}
}
}
int ans=0;
for(int i=0;i<m;++i)
{
ans=max(ans,dp[n-1][i]);
}
printf("%d\n",ans);
}
return 0;
}