[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
",all-flow());
	return 0;
}

总结

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