BZOJ 1412: [ZJOI2009]狼和羊的故事

题目地址：http://www.lydsy.com/JudgeOnline/problem.php?id=1412

题目大意：将一个矩阵中的1和2分隔，求出最小的分隔代价。

算法讨论：

        很明显这是一个类二分图，由于要将狼和羊分开，所以一边是狼，一边是羊。

        然后求这个图的最小割。

        相邻的格子之间连容量为1的边，含义是如果割这条边需要1的代价。

        源点向每只狼连容量为无穷大的边，每只羊向汇点连容量为容量为无穷大的边。

        然后用Dinic算法求最大流。

Code：

/*
 * Problem:1412
 * Author:PYC
 */

#include <cstdio>
#include <queue>
#include <algorithm>

#define maxn 1000000
#define N 1000
#define oo 1000000000

using namespace std;

int n,m,s,t,mm,tot,next[maxn+1],son[maxn+1],ed[maxn+1],cap[maxn+1],opp[maxn+1],d[maxn+1],ws[N+1][N+1],id[N+1][N+1];

void add(int x,int y,int z){
	next[++mm]=son[x],son[x]=mm,ed[mm]=y,cap[mm]=z;
}

bool build(){
	queue<int> Q;
	for (int i=1;i<=t;++i) d[i]=-1;
	d[s]=0;
	Q.push(s);
	while (!Q.empty()){
		int x=Q.front();Q.pop();
		for (int i=son[x];i;i=next[i]){
			int y=ed[i];
			if (cap[i] && d[y]==-1){
				d[y]=d[x]+1;
				if (y==t) return 1;
				Q.push(y);
			}
		}
	}
	return 0;
}

int find(int x,int flow){
	if (x==t) return flow;
	int w=0,ret;
	for (int i=son[x];i && w<flow;i=next[i]){
		int y=ed[i];
		if (cap[i] && d[y]==d[x]+1 && (ret=find(y,min(flow-w,cap[i])))){
			cap[i]-=ret;
			cap[opp[i]]+=ret;
			w+=ret;
		}
	}
	if (!w) d[x]=-1;
	return w;
}

int dinic(){
	int ans=0,f;
	while (build()) while (f=find(s,oo)) ans+=f;
	return ans;
}

int main(){
	scanf("%d%d",&n,&m),s=0,t=n*m+1;
	for (int i=1;i<=n;++i) for (int j=1;j<=m;++j) scanf("%d",&ws[i][j]),id[i][j]=++tot;
	for (int i=1;i<=n;++i) for (int j=1;j<=m;++j){
		if (ws[i][j]==1) add(s,id[i][j],oo),opp[mm]=mm+1,add(id[i][j],s,0),opp[mm]=mm-1;
		if (ws[i][j]==2) add(id[i][j],t,oo),opp[mm]=mm+1,add(t,id[i][j],0),opp[mm]=mm-1;
	}
	for (int i=1;i<=n;++i) for (int j=1;j<=m;++j){
		if (i>1) add(id[i-1][j],id[i][j],1),opp[mm]=mm+1,add(id[i][j],id[i-1][j],0),opp[mm]=mm-1;
		if (i<n) add(id[i+1][j],id[i][j],1),opp[mm]=mm+1,add(id[i][j],id[i+1][j],0),opp[mm]=mm-1;
		if (j>1) add(id[i][j-1],id[i][j],1),opp[mm]=mm+1,add(id[i][j],id[i][j-1],0),opp[mm]=mm-1;
		if (j<m) add(id[i][j+1],id[i][j],1),opp[mm]=mm+1,add(id[i][j],id[i][j+1],0),opp[mm]=mm-1;
	}
	printf("%d\n",dinic());
	return 0;
}

By Charlie Pan

Mar 15,2014