BZOJ1412 [ZJOI2009]狼和羊的故事 最小割

有一个n*m的网格,其中标1的是狼,标2的是羊,标0的是空地。现在要在格子与格子之间修一些栅栏,使任意的狼与羊互不联通。n,m<=100.

割的定义是:使S集和T集不存在通路。而题目又要求建的栅栏最少,于是就是最小割问题了。

从源点向所有狼连INF的边,从所有羊向汇点连INF的边,保证狼和羊都在不同的点集里.

再从狼、空地向空地、羊连1的边.

跑最大流即可.

#include<bits/stdc++.h>
#define LL long long
#define clr(x,i) memset(x,i,sizeof(x))
using namespace std;
const int N=202,INF=1e7;
struct Edge{
	int to,nex,cap;
}edge[N*N*4];
int n,m,tot,head[N*N],cur[N*N];
inline void Add(int u,int v,int c)
{
	edge[++tot].to=v;edge[tot].nex=head[u];edge[tot].cap=c;head[u]=tot;
	edge[++tot].to=u;edge[tot].nex=head[v];edge[tot].cap=0;head[v]=tot;
}
int mp[N][N],s,t,dx[4]={0,0,1,-1},dy[4]={1,-1,0,0};
inline bool Out(int x,int y)
{
	return (x<1||x>n||y<1||y>m);
}
void work()
{
	tot=1;s=0;t=2*n*m+1;
	for(int i=1;i<=n;i++)
	{
		for(int j=1;j<=m;j++)
		{
			if(mp[i][j]==1)
			  Add(s,(i-1)*m+j,INF);
			if(mp[i][j]==2)
			  Add((i-1)*m+j,t,INF);
			for(int k=0;k<4;k++){
				int nx=i+dx[k],ny=j+dy[k];
				if(Out(nx,ny)||mp[i][j]==2)continue;
				if( !(mp[i][j]==1&&mp[nx][ny]==1) )
				  Add((i-1)*m+j,(nx-1)*m+ny,1);
			}
		}
	}
}
int dist[N*N],que[N*N];
bool bfs()
{
	clr(dist,60);que[1]=s;dist[s]=0;
	int l=0,r=1;
	while(l<r)
	{
		int u=que[++l];
		for(int i=head[u];i;i=edge[i].nex){
			int v=edge[i].to,cap=edge[i].cap;
			if(dist[v]>INF&&cap>0){
				dist[v]=dist[u]+1;
				que[++r]=v;
			}
		}
	}
	return dist[t]<INF;
}
int dfs(int u,int maxf)
{
	if(u==t)return maxf;
	for(int i=cur[u];i;i=edge[i].nex){
		cur[u]=i;
		int v=edge[i].to,cap=edge[i].cap;
		if(dist[v]==dist[u]+1&&cap>0){
			int flow=dfs(v,min(cap,maxf));
			if(flow){
				edge[i].cap-=flow;
				edge[i^1].cap+=flow;
				return flow;
			}
		}
	}
	return 0;
}
int dinic()
{
	int ret=0,flow;
	while(bfs()){
		for(int i=s;i<=t;i++)cur[i]=head[i];
		while(flow=dfs(s,INF))
		  ret+=flow;
	}
	return ret;
}
int main()
{
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;i++)
	  for(int j=1;j<=m;j++)
	    scanf("%d",&mp[i][j]);
	work();
	int ans=dinic();
	printf("%d",ans);
	return 0;
}




发布了39 篇原创文章 · 获赞 4 · 访问量 5816

猜你喜欢

转载自blog.csdn.net/Wolf_Reiser/article/details/78597617
今日推荐