Codeforces 1592F2 Alice and Recoloring 2

有一张 \(n \times m\) 的方格图,初始全白,目标状态给定。

有 4 种操作(反转的意思是黑 \(\to\) 白,白 \(\to\) 黑):

  • 反转一个包含 \((1, 1)\) 的子矩形,花费 \(1\)
  • 反转一个包含 \((n, 1)\) 的子矩形,花费 \(3\)
  • 反转一个包含 \((1, m)\) 的子矩形,花费 \(4\)
  • 反转一个包含 \((n, m)\) 的子矩形,花费 \(2\)

求达成目标状态的最小花费。

\(1 \le n, m \le 500\)

2s, 256MB


引理 1:第 2 和第 3 种操作不会用到。

证明:考虑到不管是第 2 种还是第 3 种操作,都可以用两次第 1 种操作代替,而且代价显然更优。所以永远不会使用第 2 和第 3 种操作。

现在的问题就转化为了只存在第 1 和第 4 种操作的情况了。

将目标状态与初始状态对换,即给定一个初始有黑白两色的方格图,花最小的代价使得全部变成白色。

矩形反转显然是不好处理的,考虑弄一个类似前缀和的东西来优化掉。

将黑色视为 \(1\),白色视为 \(0\)。构造一个数组 \(a\),其中 \(a_{i, j} = s_{i, j} \oplus s_{i + 1, j} \oplus s_{i, j + 1} \oplus s_{i + 1, j + 1}\)(超出网格的部分默认是白色)。

非常显然,当 \(a\) 数组全部变为 \(0\) 时,\(s\) 数组也就全部变为了 \(0\)

观察 1, 4 两种操作对 \(a\) 数组的影响,发现是:

  • 对于第 1 种操作,只会有 1 个格子的 \(a\) 发生了反转。
  • 对于第 4 种操作,会有 4 个格子的 \(a\) 发生反转,且这 4 个格子形如 \((x, y)\)\((n, y)\)\((x, m)\)\((n, m)\)。记这样的操作为 \(op(x, y)\)

引理 2:不会同时使用 \(op(x, y_1)\)\(op(x, y_2)\)。同理不会同时使用 \(op(x_1, y)\)\(op(x_2, y)\)

证明:以前一种为例,\((n, m)\)\((x, m)\) 都被反转了两次,所以不会发生改变。那么也就是花费了 \(4\) 的代价来反转了 \(2\times 2=4\) 个格子。这显然可以被第 1 种操作代替。

引理 3:除非 \((x, y)\)\((n, y)\)\((x, m)\) 都为 \(1\),才会使用 \(op(x, y)\)

证明:如果这中间有一个不为 \(1\),那么这次操作就产生了一个错误的反转。为了达成最终目标,显然会使用一次第 1 种操作把它反转回来(注:不会是另一个第 4 种操作,根据引理 2 可以得知)。那么仅仅是反转了另外 3 个格子,代价都至少为 \(1 + 2 = 3\),完全可以使用第 1 种操作代替。

于是就可以做题了,建立一个二分图,左部有 \(n - 1\) 个点代表行,右部有 \(m - 1\) 个点代表列。

对于 \((x, y)\),如果它满足引理 3 的条件,则把左部的 \(x\) 和右部的 \(y\) 连边。

求这个二分图的最大匹配数 \(k\) 即可。答案为 \(\textit{rem} - k\)\(\textit{rem}\) 表示剩下的 \(1\) 的个数。

我使用的是 dinic 求二分图最大匹配,时间复杂度为 \(O(n^2 \sqrt{n})\)

#include <bits/stdc++.h>
const int N = 505;
char str[N][N];
int n, m, cnt = -1, a[N][N], h[N << 1], S, T;

struct edge {
	int v, w, nxt;
} e[N * N * 2];

void add_edge(int u, int v) {
	e[++cnt] = (edge){v, 1, h[u]}; h[u] = cnt;
	e[++cnt] = (edge){u, 0, h[v]}; h[v] = cnt;
}

int que[N << 1], hd, tl, lev[N << 1];
 
bool bfs() {	 
	memset(lev, 0, sizeof(lev));
	hd = tl = lev[S] = 1; que[1] = S;
	while(hd <= tl) {
		int u = que[hd++];
		for(int i = h[u]; ~i; i = e[i].nxt) {
			int v = e[i].v;
			if(!lev[v] && e[i].w > 0) {
				que[++tl] = v;
				lev[v] = lev[u] + 1;
			}
		}
	}
	return lev[T];
}

int cur[N << 1];
 
int dfs(int u, int can_flow) {
	if(u == T) return can_flow;
	int res_flow = 0;
	for(int &i = cur[u]; ~i; i = e[i].nxt) {
		int v = e[i].v;
		if(lev[v] == lev[u] + 1 && e[i].w > 0) {
			int will_flow = dfs(v, std::min(can_flow, e[i].w));
			res_flow += will_flow;
			can_flow -= will_flow;
			e[i ^ 1].w += will_flow;
			e[i].w -= will_flow;
			if(!can_flow) break;
		}
	}
	if(!res_flow) lev[u] = 0;
	return res_flow;
}
 
int dinic() {
	int res = 0; 
	while(bfs()) {
		memcpy(cur, h, sizeof(h));
		res += dfs(S, INT_MAX);
	}
	return res;
}

int main() {
	memset(h, -1, sizeof(h));

	scanf("%d %d", &n, &m); 
	S = n + m + 1; T = n + m + 2;
	for(int i = 1; i <= n; i++) {
		scanf("%s", str[i] + 1);
	}
	for(int i = 1; i <= n; i++) {
		for(int j = 1; j <= m; j++) {
			a[i][j] = (str[i][j] == 'B') ^ (str[i][j + 1] == 'B') ^ (str[i + 1][j] == 'B') ^ (str[i + 1][j + 1] == 'B');
		}
	}
	for(int i = 1; i < n; i++) {
		for(int j = 1; j < m; j++) {
			if(a[i][j] && a[n][j] && a[i][m]) {
				add_edge(i, j + n);
			}
		}
	}
	for(int i = 1; i < n; i++) add_edge(S, i);
	for(int j = 1; j < m; j++) add_edge(j + n, T);

	int k = dinic(), ans = 0;
	
	a[n][m] ^= (k & 1);
	for(int i = 1; i <= n; i++) {
		for(int j = 1; j <= m; j++) {
			ans += a[i][j];
		}
	}
	ans -= k;
	printf("%d\n", ans);
	return 0;
}
posted @ 2021-10-07 17:27  syksykCCC  阅读(135)  评论(0编辑  收藏  举报