Hatsune Miku（简单DP）

题目链接
题目大意
给你一个长度为n的数组p，对于每个不为-1的元素pi，都满足1<=pi<=m, 当pi等于-1时pi可以修改为1到m的任意一个数字，给你一个大小为m*m的二维数组a，对于每个数组p，都有一个属于自己的值，等于（a[p[1]][p[2]] + a[p[2]][p[3]] + … + a[p[n - 1]][p[n]]）;
让你求此值最大是多少；

解题思路

本题数据比较小，可以写出来当数组长度为i时，前i位可能出现的所有值，具体方法可以通过枚举最后一个值的坐标为多少来实现；

#include<bits/stdc++.h>

const int N = 110;

using namespace std;

int dp[N][N], a[N][N];
int p[N];

int main() {
	int t;
	cin >> t;
	while(t--) {
		int n, m;
		cin >> n >> m;
		
		for (int i = 1; i <= m; i ++) 
		    for (int j = 1; j <= m; j ++) cin >> a[i][j];
		for(int i = 1; i <= n; i ++) cin >> p[i];
		
		memset(dp, 0, sizeof dp);	
		
		for(int i = 2; i <= n; i++) //枚举数组长度
			for(int j = 1; j <= m; j ++)
				for(int k = 1; k <= m; k ++) { //枚举最后一个值的坐标j, k；
					if((p[i - 1] == -1 || j == p[i - 1]) && (p[i] == -1 || p[i] == k)) //判断该坐标是否能出现；
					    dp[i][k] = max(dp[i][k], dp[i - 1][j] + a[j][k]);												
				}
		
		int maxx = 0;
		for(int i = 1; i <= m; i ++)  maxx = max(maxx, dp[n][i]);
		
		cout << maxx << endl;
		
	}
}