题目地址: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