一开始想不到是用动态规划来做,网上搜了一下才发现要用dp
首先很容易想到dp定义的状态是dp[i],表示前i个数的最大值
但是很快发现由dp[i-1]并不能得出dp[i]
可以考虑3个的情形,假设第一个是确定的1,我第二个选择第一行里最大的假设是5
第三个选择第5行里最大的假设是2,则和是7
但是我第二个可以选择第一行里不是那么大的,假设是4,第三个选择第4行李最大的假设是5,则和是9
也就是说dp[2]显然是6,但是我并不能从6(1,5)来推出9,显然还要有其他信息,这个信息就是放上去的这个数字,所以设dp[i][j]表示前i个note最后一个放j的和的最大值
分4种情况递推一下
#include <iostream>
#include<stdio.h>
#include<algorithm>
#include<string.h>
#include<math.h>
#define ENTER printf("\n");
#define ll long long
using namespace std;
int t;
int n, m;
int score[50 + 1][50 + 1];
int dp[100 + 1][50 + 1];
int num[100 + 1];
int main()
{
scanf("%d", &t);
while (t--)
{
scanf("%d%d", &n, &m);
for (int i = 1; i <= m; i++)
{
for (int j = 1; j <= m; j++)
{
scanf("%d", &score[i][j]);
}
}
for (int i = 0; i < n; i++)
{
scanf("%d", num + i+1);
}
memset(dp, 0, sizeof(dp));
for (int j = 1; j <= m; j++)
{
dp[1][j] = 0;
}
for (int i = 2; i <= n; i++)
{
for (int j = 1; j <= m; j++)
{
if (num[i - 1] < 0 && (num[i] < 0 || j == num[i]))
{
for (int k = 1; k <= m; k++)
{
dp[i][j] = max(dp[i][j], dp[i - 1][k] + score[k][j]);
}
}
else if (num[i - 1] > 0 && num[i] < 0)
{
dp[i][j] = dp[i - 1][num[i - 1]] + score[num[i - 1]][j];
}
else
{
if (j == num[i])
{
dp[i][j] = dp[i - 1][num[i - 1]] + score[num[i - 1]][num[i]];
}
}
}
}
int maxx = -1;
for (int i = 1; i <= m; i++)
{
maxx = max(dp[n][i], maxx);
}
printf("%d\n", maxx);
}
return 0;
}