CF79D Password （差分+状压 dp+最短路/bfs）

CF79D Password

差分 + 状压 dp + 最短路/bfs

好题。区间取反不好做？考虑差分一下，那么操作就转化为 $i$ 和 $i+k$ 两个位置的单点取反。因为差分数组上 $1$ 的数量 $\le 20$，满足题目状态等价于满足差分数组的状态，考虑状压 $1$ 的位置。设 $f_s$ 表示达到状态 $s$ 的最少操作数。

考虑转移。实际上可以把所有操作拆分成若干条从其中一个 $1$ 到另一个 $1$ 的路径（这里的 $1$ 指目标状态中的 $1$）。考虑建图，对于长度 $v$ 在每个位置连边 $(i,i+v)$。那么显然每条路径要最短，跑大约 $20$ 次单源最短路/bfs 就可以了。那么枚举两个位置 $i$ 和 $j$，有$f_{s|2^i|2^j}=min(f_s+m{p}_{i,j})$。

复杂度 $O(2^{2k}k)$。

#include <bits/stdc++.h>
#define pii std::pair<int, int>
#define fi first
#define se second
#define pb push_back

using i64 = long long;
using ull = unsigned long long;
const i64 iinf = 0x3f3f3f3f, linf = 0x3f3f3f3f3f3f3f3f;
const int N = 1e4 + 10, S = 21;
int n, k, l, num;
int a[N], dis[25][N], vis[N], id[N], f[1 << S];
std::vector<int> e[N];
struct com {
	int v, w;
	friend bool operator < (com a, com b) {
		return a.w > b.w;
	}
} tmp;
void dij(int s) {
	id[++num] = s;
	std::priority_queue<com> q;
	for(int i = 1; i <= n + 1; i++) dis[num][i] = iinf, vis[i] = 0;
	dis[num][s] = 0;
	tmp = {s, 0};
	q.push(tmp);
	while(!q.empty()) {
		int u = q.top().v;
		q.pop();
		if(vis[u]) continue;
		vis[u] = 1;
		for(auto v : e[u]) {
			if(dis[num][v] > dis[num][u] + 1) {
				dis[num][v] = dis[num][u] + 1;
				tmp = {v, dis[num][v]};
				q.push(tmp);
			}
		}
	}
	return;
}
int main() {
    std::ios::sync_with_stdio(false);
    std::cin.tie(nullptr);
    
	std::cin >> n >> k >> l;
	for(int i = 1; i <= k; i++) {
		int x;
		std::cin >> x;
		a[x] = 1;
	}
	
	for(int i = 1; i <= l; i++) {
		int v;
		std::cin >> v;
		for(int j = 1; j <= n + 1; j++) {
			if(j + v > n + 1) continue;
			e[j].pb(j + v);
			e[j + v].pb(j);
		}
	}

	int cnt = 0;
	for(int i = n + 1; i >= 1; i--) {
		a[i] = abs(a[i] - a[i - 1]);
		cnt += a[i];
	}
	for(int i = 1; i <= n + 1; i++) if(a[i]) dij(i);

	memset(f, 0x3f, sizeof(f));
	f[0] = 0;
	int lim = (1 << cnt) - 1;
	for(int s = 0; s <= lim; s++) {
		for(int i = 0; i < cnt; i++) {
			if((s >> i) & 1) continue;
			for(int j = i + 1; j < cnt; j++) {
				if((s >> j) & 1) continue;

				f[s | (1 << i) | (1 << j)] = std::min(f[s | (1 << i) | (1 << j)], f[s] + dis[i + 1][id[j + 1]]);

			}
		}
	}
	std::cout << (f[lim] == iinf ? -1 : f[lim]) << "\n";

	return 0;
}