[JZOJ1900] 【2010集训队出题】矩阵

题目

题目大意

题目化简一下，就变成：
构造一个\(01\)数列\(A\)，使得\(D=\sum A_iA_jB_{i,j}-\sum A_iC_i\)最大。
问这个最大的\(D\)是多少。

---

正解

其实这是一个网络流的二元关系问题……
如果\(A_i\)为\(1\)，则会有\(-C_i\)的贡献。
如果\(A_i\)和\(A_j\)皆为\(1\)，则会有\(B_{i,j}\)的贡献。
然后很显然地，\(70\)分的方法就出来了：每个点朝汇点连一条容量为\(C_i\)的边，对于每个\(B_{i,j}\)，建一个新点，从源点朝它连一条容量为\(B_{i,j}\)的边，它朝\(i\)和\(j\)连容量为无限大的边。然后最小割即可。
这个算法的瓶颈在于这些新点太多了，能不能不用建立新点？
实际上有个很妙的方法：对于每一对\(i\)和\(j\)，从原点向\(i\)连一条容量为\(B_{i,j}\)的边，同样地向\(j\)连一条容量为\(B_{j,i}\)的边。\(i\)向\(j\)连一条容量为\(B_{i,j}\)的边，\(j\)向\(i\)连一条容量为\(B_{j,i}\)的边。
那么这有什么用呢？当\(C_i\)的那条边被保留的时候，源点向\(i\)连的那条\(B_{i,j}\)的边会被割掉，还有源点连向\(j\)或者\(j\)连向\(i\)的那条边也会被割掉。

另一种建图方式跟这个比较类似，只是把边权换成了\(\frac{B_{i,j}+B_{j,i}}{2}\)罢了。因为只要保留\(C_i\)或者\(C_j\)，割掉的边都是\(B_{i,j}+B_{j,i}\)。
对于源点向\(i\)和\(j\)连的边，显然可以合并起来。所以图中的点和边的数量就大大减少了。

---

代码

using namespace std;
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <climits>
#define N 610
inline int input(){
	char ch=getchar();
	while (ch<'0' || '9'<ch)
		ch=getchar();
	int x=0;
	do{
		x=x*10+ch-'0';
		ch=getchar();
	}
	while ('0'<=ch && ch<='9');
	return x;
}	
int n;
int b[N][N],c[N];
struct EDGE{
	int to,c;
	EDGE *las;	
} e[2000000];
int ne;
EDGE *last[N];
inline void link(int u,int v,int c){
	e[ne]={v,c,last[u]};
	last[u]=e+ne++;
}
int S,T;
#define rev(ei) (e+(((ei)-e)^1))
int dis[N],gap[N],BZ;
EDGE *cur[N];
int dfs(int x,int s){
	if (x==T)
		return s;
	int have=0;
	for (EDGE *ei=cur[x];ei;ei=ei->las){
		cur[x]=ei;
		if (ei->c && dis[x]==dis[ei->to]+1){
			int t=dfs(ei->to,min(s-have,ei->c));
			ei->c-=t,rev(ei)->c+=t,have+=t;
			if (have==s)
				return s;
		}
	}
	cur[x]=last[x];
	if (!--gap[dis[x]])
		BZ=0;
	dis[x]++;
	gap[dis[x]]++;
	return have;
}
inline int flow(){
	gap[0]=n+2;
	int res=0;
	BZ=1;
	while (BZ)
		res+=dfs(S,INT_MAX);
	return res;
}
int main(){
	n=input();
	for (int i=1;i<=n;++i)
		for (int j=1;j<=n;++j)
			b[i][j]=input();
	for (int i=1;i<=n;++i)
		c[i]=input();
	S=n+1,T=n+2;
	int all=0;
	for (int i=1;i<=n;++i){
		int sum=0;
		for (int j=1;j<=n;++j)
			sum+=b[j][i];
		all+=sum;
		link(S,i,sum),link(i,S,0);
		for (int j=1;j<i;++j)
			link(i,j,b[j][i]),link(j,i,b[i][j]);
		link(i,T,c[i]),link(T,i,0);
	}
	printf("%d\n",all-flow());
	return 0;
}

总结

见到二元关系类型的题目，首先要想到网络流啊……