洛谷 P2598 [ZJOI2009]狼和羊的故事

题目描述

“狼爱上羊啊爱的疯狂，谁让他们真爱了一场；狼爱上羊啊并不荒唐，他们说有爱就有方向．．．．．．” Orez听到这首歌，心想：狼和羊如此和谐，为什么不尝试羊狼合养呢？说干就干！ Orez的羊狼圈可以看作一个n*m个矩阵格子，这个矩阵的边缘已经装上了篱笆。可是Drake很快发现狼再怎么也是狼，它们总是对羊垂涎三尺，那首歌只不过是一个动人的传说而已。所以Orez决定在羊狼圈中再加入一些篱笆，还是要将羊狼分开来养。 通过仔细观察，Orez发现狼和羊都有属于自己领地，若狼和羊们不能呆在自己的领地，那它们就会变得非常暴躁，不利于他们的成长。 Orez想要添加篱笆的尽可能的短。当然这个篱笆首先得保证不能改变狼羊的所属领地，再就是篱笆必须修筑完整，也就是说必须修建在单位格子的边界上并且不能只修建一部分。

输入输出格式

输入格式：

文件的第一行包含两个整数n和m。接下来n行每行m个整数，1表示该格子属于狼的领地，2表示属于羊的领地，0表示该格子不是任何一只动物的领地。

输出格式：

文件中仅包含一个整数ans，代表篱笆的最短长度。

输入输出样例

输入样例#1：

2 2
2 2 
1 1 

输出样例#1：

2

说明

数据范围

10%的数据 n，m≤3

30%的数据 n，m≤20

100%的数据 n，m≤100

将源点与狼连容量为inf的边，将羊与汇点连容量为inf的边，将狼和其他，其他和羊，狼和羊连容量为1的边。将相同的动物连容量为1的双向边，求最小割。

#include<iostream>
#include<cstring>
#include<cstdio>
#include<queue>
using namespace std;
const int N=105;
const int inf=1e9+7;
int n,m,s,t,mct=1,ans,mp[N][N],hd[N*N],g[N][N],pre[N*N];
queue<int>q;
struct edge
{
	int to,nxt,val;
}v[10*N*N];
void addedge(int x,int y,int z)
{
	v[++mct].to=y;
	v[mct].val=z;
	v[mct].nxt=hd[x];
	hd[x]=mct;
}
bool bfs()
{
	memset(pre,0,sizeof(pre));
	pre[s]=1;
	q.push(s);
	while(!q.empty())
	{
		int u=q.front();
		q.pop();
		for(int i=hd[u];i;i=v[i].nxt)
			if(v[i].val&&!pre[v[i].to])
			{
				pre[v[i].to]=pre[u]+1;
				q.push(v[i].to);
			}
	}
	return pre[t];
}
int dfs(int u,int lft)
{
	if(u==t||lft==0)
		return lft;
	int r=lft;
	for(int i=hd[u];i;i=v[i].nxt)
		if(r&&v[i].val&&pre[v[i].to]==pre[u]+1)
		{
			int w=dfs(v[i].to,min(r,v[i].val));
			v[i].val-=w;
			v[i^1].val+=w;
			r-=w;
			if(!r)
				return lft;
		}
	if(lft==r)
		pre[u]=0;
	return lft-r;
}
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++)
			g[i][j]=(i-1)*m+j;
	for(int i=1;i<=n;i++)
		for(int j=1;j<=m;j++)
		{
			scanf("%d",&mp[i][j]);
			if(mp[i][j]==1)
				addedge(s,g[i][j],inf),addedge(g[i][j],s,0);
			if(mp[i][j]==2)
				addedge(g[i][j],t,inf),addedge(t,g[i][j],0);
		}
	for(int i=1;i<=n-1;i++)
		for(int j=1;j<=m;j++)
		{
			if((mp[i][j]==1&&(mp[i+1][j]==0||mp[i+1][j]==2))||(mp[i][j]==0&&mp[i+1][j]==2))
				addedge(g[i][j],g[i+1][j],1),addedge(g[i+1][j],g[i][j],0);
			if((mp[i+1][j]==1&&(mp[i][j]==0||mp[i][j]==2))||(mp[i+1][j]==0&&mp[i][j]==2))
				addedge(g[i+1][j],g[i][j],1),addedge(g[i][j],g[i+1][j],0);
			if(mp[i+1][j]==mp[i][j])
				addedge(g[i+1][j],g[i][j],1),addedge(g[i][j],g[i+1][j],1);
		}
	for(int i=1;i<=n;i++)
		for(int j=1;j<=m-1;j++)
		{
			if((mp[i][j]==1&&(mp[i][j+1]==0||mp[i][j+1]==2))||(mp[i][j]==0&&mp[i][j+1]==2))
				addedge(g[i][j],g[i][j+1],1),addedge(g[i][j+1],g[i][j],0);
			if((mp[i][j+1]==1&&(mp[i][j]==0||mp[i][j]==2))||(mp[i][j+1]==0&&mp[i][j]==2))
				addedge(g[i][j+1],g[i][j],1),addedge(g[i][j],g[i][j+1],0);
			if(mp[i][j]==mp[i][j+1])
				addedge(g[i][j+1],g[i][j],1),addedge(g[i][j],g[i][j+1],1);
		}
	while(bfs())
		ans+=dfs(s,inf);
	printf("%d\n",ans);
	return 0;
}