bzoj 4565 [Haoi2016]字符合并 状压dp+区间dp

题面

题目传送门

解法

• 看到$K\le 8$，应该会比较直观地想到状压dp
• 考虑一段区间，我们一定是把这段区间合并到不能继续合并为止，这样一定是最优的
• 那么，我们可以设计一下状态：$f[i][j][S]$表示将区间$[i,j]$合并成二进制数$S$的最大价值
• 考虑如何枚举中间的分界点$k$，显然是当且仅当区间$[k,j]$这段区间合并之后恰好为$S$的末尾几位
• 考虑这样做的正确性，我们在dp的时候一定做的都是正确合法的方案，而每一个正确合法的方案也一定会被dp到
• 时间复杂度：$O(n^3{2}^K)$

【注意事项】

• 请注意常数因子对程序效率带来的影响，即每一次转移的时候尽量使用那些合法的状态
• 在区间$[i,j]$恰好能合并成一个数的时候需要特别处理一下，注意不能直接用$f$数组来迭代，否则可能会出现使用本来不合法的状态来更新其他的状态的情况

代码

#include <bits/stdc++.h>
#define LL long long
#define inf 1 << 30
#define N 310
using namespace std;
template <typename node> void chkmax(node &x, node y) {x = max(x, y);}
template <typename node> void chkmin(node &x, node y) {x = min(x, y);}
template <typename node> void read(node &x) {
	x = 0; int f = 1; char c = getchar();
	while (!isdigit(c)) {if (c == '-') f = -1; c = getchar();}
	while (isdigit(c)) x = x * 10 + c - '0', c = getchar(); x *= f;
}
int a[N], c[1 << 8];
LL w[1 << 8], f[N][N][1 << 8];
int main() {
	int n, K; read(n), read(K);
	for (int i = 1; i <= n; i++) {
		char c = getchar();
		while (!isdigit(c)) c = getchar();
		a[i] = c - '0';
	}
	for (int i = 0; i < (1 << K); i++) read(c[i]), read(w[i]);
	for (int i = 1; i <= n; i++)
		for (int j = i; j <= n; j++)
			for (int S = 0; S < (1 << K); S++)
				f[i][j][S] = -inf;
	for (int i = 1; i <= n; i++) f[i][i][a[i]] = 0;
	for (int l = 2; l <= n; l++) {
		for (int i = 1; i <= n - l + 1; i++) {
			int j = i + l - 1, tmp = l;
			while (tmp >= K) tmp -= K - 1;
			for (int k = j; k > i; k -= K - 1) {
				for (int S = 0; S < (1 << K); S++) {
					if (f[i][k - 1][S] < 0) continue;
					if (f[k][j][0] >= 0) chkmax(f[i][j][S << 1], f[i][k - 1][S] + f[k][j][0]);
					if (f[k][j][1] >= 0) chkmax(f[i][j][S << 1 | 1], f[i][k - 1][S] + f[k][j][1]);
				}
			}
			if (tmp == 1) {
				LL mx[2] = {-inf, -inf};
				for (int S = 0; S < (1 << K); S++)
					if (f[i][j][S] >= 0) chkmax(mx[c[S]], f[i][j][S] + w[S]);
				f[i][j][0] = mx[0], f[i][j][1] = mx[1];
			}
		}
	}
	LL ans = 0;
	for (int S = 0; S < (1 << K); S++)
		chkmax(ans, f[1][n][S]);
	cout << ans << "\n";
	return 0;
}