【JZOJ1900】【2010集训队出题】矩阵 （经典2-SAT问题）

题意

有一个(n*n)的矩阵(B)，一个(1*n)的矩阵(C)，你现在要构造一个(1*n)的(0/1)矩阵，令((A*B-C)*A^T=D)，(D)只有一个元素，你要使得这个元素值最大。
(nleq 600)

分析

推一下矩阵乘法的式子就能转换为这样的问题：

有(n)个元素编号为(1 sim n)，选择(i)将获得(-c_i)的贡献，同时选择(i,j)将获得(b_{i,j})的贡献，找出一种方案使得贡献最大。

这是一个典型的2-SAT问题，建模方法如下：

建立源点(S)，向中间一排(n)个点连边，边权为(sum b_{i,k})，这些边存在的意义是(a_i=1)。中间一排点向汇点(T)连边，边权为(c_i)，这些边存在的意义是(a_i=0)。中间的点两两连边，边权为(b_{i,j})。

思考这个网络的最小割的意义。对于中间某个点(i)，显然((S,i))和((i,T))两条边有且仅有一条被割，割((S,i))意义为(a_i=0)，割((i,T))意义为(a_i=1)。当(a_i,a_j)有一个为(0)时，必定存在(S,i,j,T)的路径，(b_{i,j})这条边就会割掉，贡献就会减去。综上，这个网络的最小割就对应原模型的解。

Code

#include <cstdio>
#include <cstring>

const int N = 617, INF = 2147483647;
inline int read()
{
	int x = 0, f = 0;
	char c = getchar();
	for (; c < '0' || c > '9'; c = getchar()) if (c == '-') f = 1;
	for (; c >= '0' && c <= '9'; c = getchar()) x = (x << 1) + (x << 3) + (c ^ '0');
	return f ? -x : x;
}
int min(int a, int b) { return a < b ? a : b; }

int n;
int ans, b[N][N], c[N];

int tot = 1, st[N], to[N * N * 4], nx[N * N * 4], len[N * N * 4], gap[N];
int cur[N];
void add(int u, int v, int w)
{
	to[++tot] = v, nx[tot] = st[u], len[tot] = w, st[u] = tot;
	to[++tot] = u, nx[tot] = st[v], len[tot] = 0, st[v] = tot;
}

int S, T, h, t, q[N], dep[N];
void bfs()
{
	memset(dep, -1, sizeof(dep));
	memset(gap, 0, sizeof(gap));
	h = 1, q[t = 1] = T, dep[T] = 0;
	while (h <= t)
	{
		int u = q[h++];
		++gap[dep[u]];
		for (int i = st[u]; i; i = nx[i]) if (dep[to[i]] == -1) dep[to[i]] = dep[u] + 1, q[++t] = to[i];
	}
}
int dinic(int u, int flow)
{
	if (u == T) return flow;
	int rest = flow, tmp;
	for (int i = cur[u]; i; i = nx[i])
	{
		cur[u] = i;
		if (len[i] > 0 && dep[u] == dep[to[i]] + 1)
		{
			tmp = dinic(to[i], min(rest, len[i]));
			len[i] -= tmp, len[i ^ 1] += tmp, rest -= tmp;
			if (!rest) return flow;
		}
	}
	--gap[dep[u]];
	if (gap[dep[u]] == 0) dep[S] = n + 3;
	++dep[u];
	++gap[dep[u]];
	return flow - rest;
}

int main()
{
	//freopen("matrix.in", "r", stdin);
	n = read();
	for (int i = 1; i <= n; i++) for (int j = 1; j <= n; j++) b[i][j] = read(), ans += b[i][j];
	for (int i = 1; i <= n; i++) c[i] = read();
	S = 0, T = n + 1;
	for (int i = 1; i <= n; i++)
	{
		int s = 0;
		for (int j = 1; j <= n; j++) s += b[i][j];
		add(S, i, s), add(i, T, c[i]);
	}
	for (int i = 1; i <= n; i++) for (int j = 1; j <= n; j++) if (i != j) add(i, j, b[i][j]);
	bfs();
	while (dep[S] < n + 2) memcpy(cur, st, sizeof(st)), ans -= dinic(S, INF);
	printf("%d
", ans);
	return 0;
}