【BZOJ3508】开灯

【BZOJ3508】开灯

题面

bzoj

题解

其实变为目标操作和从目标操作变回来没有区别，我们考虑从目标操作变回来。

区间整体翻转（\(\text{Xor}\;1\)）有点难受，我们考虑将这个操作放在差分数组上，也就是说令\(a\)为原数组，\(c\)为差分数组，\(c_i=a_{i-1}\text{Xor}\;a_i\)。

那么我们就相当于让差分数组上的数全变为\(0\)，而一次操作就相当于让一对\(1\)消掉或一对\(0,1\)位置互换。

而两个\(1\)在其他位置消掉和在某个\(1\)的位置消掉是没有区别的，多个\(1\)也一样，所以我们可以直接\(bfs\)预处理每对点消掉的贡献。

而差分数组最多\(2K\)个\(1\)，状压每个点有没有被消掉即可。

代码

#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <cmath>
#include <algorithm>
#include <queue> 
using namespace std;

inline int gi() {
    register int data = 0, w = 1;
    register char ch = 0;
    while (ch != '-' && (ch > '9' || ch < '0')) ch = getchar();
    if (ch == '-') w = -1 , ch = getchar();
    while (ch >= '0' && ch <= '9') data = data * 10 + (ch ^ 48), ch = getchar();
    return w * data;
} 
const int INF = 0x3f3f3f3f; 
const int MAX_N = 40005, MAX_M = 105; 
int N, K, M; 
int b[MAX_M]; 
bool used[MAX_N], g[MAX_N]; 
int st[30], tp;
int cost[30][30], dep[MAX_N]; 

queue<int> que; 
void bfs(int s, int *dis) { 
	memset(dep, 0, sizeof(dep));
	dep[s] = 1;
	que.push(s);
	while (!que.empty()) {
		int x = que.front(); que.pop();
		for (int i = 1; i <= M; i++) {
			if (x + b[i] <= N && !dep[x + b[i]]) dep[x + b[i]] = dep[x] + 1, que.push(x + b[i]); 
			if (x - b[i] >= 1 && !dep[x - b[i]]) dep[x - b[i]] = dep[x] + 1, que.push(x - b[i]); 
		} 
	} 
	for (int i = 0; i < tp; i++) dis[i] = dep[st[i]] - 1; 
} 
int f[1 << 20]; 
int main () {
#ifndef ONLINE_JUDGE 
	freopen("cpp.in", "r", stdin); 
#endif 
	int Q = gi(); 
	while (Q--) { 
		memset(used, 0, sizeof(used)); tp = 0; 
		N = gi() + 1, K = gi(), M = gi(); 
		for (int i = 1; i <= K; i++) used[gi()] = 1; 
		for (int i = 1; i <= M; i++) b[i] = gi(); 
		for (int i = 1; i <= N; i++) g[i] = used[i] ^ used[i - 1]; 
		for (int i = 1; i <= N; i++) if (g[i]) st[tp++] = i; 
		for (int i = 0; i < tp; i++) bfs(st[i], cost[i]); 
		memset(f, 0x3f, sizeof(f)); 
		f[0] = 0; 
		for (int S = 0; S < (1 << tp) - 1; S++) { 
			for (int i = 0; i < tp; i++) { 
				if (S >> i & 1) continue; 
				for (int x = i + 1; x < tp; x++) { 
					if ((S >> x & 1) == 0 && (cost[i][x] != -1)) { 
						int tmp = (S | (1 << i) | (1 << x)); 
						f[tmp] = min(f[tmp], f[S] + cost[i][x]); 
					} 
				} 
				break; 
			} 
		} 
		printf("%d\n", f[(1 << tp) - 1] == INF ? -1 : f[(1 << tp) - 1]);
	} 
	return 0; 
}