题目链接:http://acm.hdu.edu.cn/showproblem.php?pid=3046
题目大意:在n*m的格子内用最短的围墙将狼和羊分开。
题目思路:最小割。我的做法是把狼和羊分别缩点,也可以增加虚拟源点和汇点。
#include<stdio.h>
#include<stdlib.h>
#include<string.h>
#include<string>
#include<queue>
#include<algorithm>
#include<vector>
#include<stack>
#include<list>
#include<iostream>
#include<map>
using namespace std;
#define inf 0x7f3f3f3f
#define Max 50000
int max(int a,int b)
{
return a>b?a:b;
}
int min(int a,int b)
{
return a<b?a:b;
}
int dis[Max],gap[Max],pre[Max],cur[Max],p[Max];
int d[4][2]={0,1,1,0,0,-1,-1,0};
int mp[220][220];
int n,m,s,t,eid;
struct node
{
int to,next,c;
}e[8*Max];
void addedge(int u,int v,int c)
{
e[eid].to=v;
e[eid].c=c;
e[eid].next=p[u];
p[u]=eid++;
}
void ISAP(int st,int ed,int n,int count) ///起点,终点,顶点数
{
memset(dis, 0, sizeof(dis));
memset(gap, 0, sizeof(gap)); gap[0]=n;
memcpy(cur, p, sizeof(p)); ///memcpy!
int i,flag,v,u=pre[st]=st,maxflow=0,aug=inf; //puts("akk");
while(dis[st] < n)
{
for(flag=0,i=cur[u];i!=-1; i=e[i].next) /// cur[u]
if(e[i].c&& dis[u] == dis[e[i].to]+1)
{
flag = 1;
break;
}
if(flag)
{
if(aug > e[i].c)
aug = e[i].c;
v = e[i].to;
pre[v] = u;
cur[u] = i;
u = v;
if(u == ed)
{
for(u=pre[u]; 1;u=pre[u]) ///notice!
{
e[cur[u]].c -= aug;
e[cur[u]^1].c += aug;
if(u==st) break;
// puts("akkk");
}
maxflow += aug;
aug = inf;
}
}
else
{
int minx = n;
for(i=p[u]; i!=-1; i=e[i].next)
if(e[i].c&& dis[e[i].to]<minx)
{
minx = dis[e[i].to];
cur[u] = i;
}
if(--gap[dis[u]] == 0)
break;
dis[u] = minx+1;
gap[dis[u]]++;
u = pre[u];
}
}
printf("Case %d:\n%d\n",count,maxflow);
}
int main()
{
int m,n,t,count=1;
int u,v,c,i,j,k,x,y;
while( scanf("%d%d",&n,&m)!=EOF)
{
eid=0;
memset(p,-1,sizeof(p));
for(i=0;i<n;i++)
for(j=0;j<m;j++)
scanf("%d",&mp[i][j]);
for(i=0;i<n;i++)
for(j=0;j<m;j++)
{
for(k=0;k<4;k++)
{
x=i+d[k][0];
y=j+d[k][1];
if(x>=0&&y>=0&&x<n&&y<m)
{
if(mp[i][j]==2&&mp[x][y]==0)
{
addedge(0,x*m+y+1,1);
addedge(x*m+j+1,0,0);
}
else if(mp[i][j]==2&&mp[x][y]==1)
{
addedge(0,n*m+1,1);
addedge(n*m+1,0,0);
}
else if(mp[i][j]==0&&mp[x][y]==1)
{
addedge(i*m+j+1,n*m+1,1);
addedge(n*m+1,i*m+j+1,0);
}
else if(mp[i][j]==0&&mp[x][y]==0)
{
addedge(i*m+j+1,x*m+y+1,1);
addedge(x*m+y+1,i*m+j+1,0);
}
}
}
}
ISAP(0,n*m+1,n*m+2,count++);
// printf("%d\n")
}
}