hdu 5074 Hatsune Miku (鞍山现场赛E题）

hdu 5074 Hatsune Miku

题意很简单，有m个符号，从1到m标号，给定一个符号的序列，若i后边紧邻着j，则得到的分数是s(i,j)，若序列中某一位是-1，则表示这一位可以选择任意符号，求出这个符号序列可以得到的最大分数。

输入数据都很小，在100的范围内，使用int就够了。第一行是n，m，分别表示序列长度和符号个数，下面是个m×m的矩阵，第i行第j列表示符号i后跟着符号j得到的分数，接下来一行有n个数，表示序列，若为-1，表示这个位置可以填任意符号。求出最大的分数。

求最优值，用dp来做，dp[i][j]表示第i位为j时候的最大分数。如果第i位已经确定，为cur，则除了dp[i][cur可能不为0外，其他都为0，若i-1位已经固定，则这一位只能取num[i-1]，就不需要枚举前一位的值了，因此可以分为四种情况，处理好就可以了。

值得注意的是，尽管0是合理的分数，但是将dp数组初始化为0，也不影响最终答案，还会便于编程。

代码如下：

/*************************************************************************
	> File Name: 5074.cpp
	> Author: gwq
	> Mail: gwq5210@qq.com 
	> Created Time: 2014年10月25日 星期六 23时55分20秒
 ************************************************************************/

#include <cmath>
#include <ctime>
#include <cctype>
#include <climits>
#include <cstdio>
#include <cstdlib>
#include <cstring>

#include <map>
#include <set>
#include <queue>
#include <stack>
#include <vector>
#include <sstream>
#include <iostream>
#include <algorithm>

#define INF (INT_MAX / 10)
#define SQR(x) ((x) * (x))
#define rep(i, n) for (int i = 0; i < (n); ++i)
#define repf(i, a, b) for (int i = (a); i <= (b); ++i)
#define repd(i, a, b) for (int i = (a); i >= (b); --i)
#define clr(arr, val) memset(arr, val, sizeof(arr))
#define pb push_back
#define sz(a) ((int)(a).size())
#define middle(x, y) ((x + y) >> 1)

using namespace std;
typedef set<int> si;
typedef vector<int> vi;
typedef map<int, int> mii;
typedef long long ll;

const double esp = 1e-5;

#define N 110

int s[N][N], dp[N][N], num[N];

int main(int argc, char *argv[])
{
	int t;

	scanf("%d", &t);
	while (t--) {
		int n, m;

		scanf("%d%d", &n, &m);

		clr(s, 0);
		clr(dp, 0);
		clr(num, 0);
		for (int i = 1; i <= m; ++i) {
			for (int j = 1; j <= m; ++j) {
				scanf("%d", &s[i][j]);
			}
		}
		for (int i = 1; i <= n; ++i) {
			scanf("%d", &num[i]);
		}

		for (int i = 2; i <= n; ++i) {
			for (int j = 1; j <= m; ++j) {
				int cur = num[i];
				//当前位可以填任意数
				if (num[i] < 0) {
					cur = j;
				}

				for (int k = 1; k <= m; ++k) {
					int pre = num[i - 1];
					//前一位可以填任意数
					if (num[i - 1] < 0) {
						pre = k;
					}
					dp[i][cur] = max(dp[i][cur], dp[i - 1][pre] + s[pre][cur]);
				}
			}
		}

		int ans = 0;
		for (int i = 1; i <= m; ++i) {
			ans = max(ans, dp[n][i]);
		}

		printf("%d\n", ans);
	}

	return 0;
}