[BZOJ1412/Luogu2598][ZJOI2009]狼和羊的故事

题目链接：

BZOJ1412

Luogu2598

题意好迷。。

一个比较简单的最小割模型。

对于所有狼与源点连边，羊与汇点连边，容量\(+\infty\)。

对于每个点向四周连边，容量为\(1\)，代表联通。

然后跑一遍最小割就行了（羊和狼联通则有\(1\)流量，相当于建栅栏）。

#include <cstdio>
#include <cstring>
#define ID(x,y) (((x)-1)*m+y)

inline int Min(int a,int b){return a<b?a:b;}
inline int Max(int a,int b){return a>b?a:b;}

int n,m,St,Ed;
int Head[10005],Next[150005],To[150005],Val[150005],En=1;
int Dis[10005],Pre[10005],Cur[10005],Cnt[10005];
const int Inf=0x3f3f3f3f,nx[]={-1,1,0,0},ny[]={0,0,-1,1};

inline void Add(int x,int y,int z)
{
    Next[++En]=Head[x],To[Head[x]=En]=y,Val[En]=z;
    Next[++En]=Head[y],To[Head[y]=En]=x,Val[En]=0;
}

int Augment()
{
    int Res=Inf;
    for(int x=Ed;x!=St;x=To[Pre[x]^1])Res=Min(Res,Val[Pre[x]]);
    for(int x=Ed,i;x!=St;x=To[i^1])Val[i=Pre[x]]-=Res,Val[i^1]+=Res;
    return Res;
}

int ISAP()
{
    memcpy(Cur,Head,sizeof Cur);
    Cnt[0]=Ed;
    int Flow=0,x=St;
    while(Dis[St]<Ed)
    {
        if(x==Ed)Flow+=Augment(),x=St;
        bool Flag=false;
        for(int i=Cur[x],y;i;i=Next[i])
            if(Val[i]&&Dis[y=To[i]]+1==Dis[x])
                Flag=true,Cur[x]=Pre[y]=i,x=y,i=0;
        if(Flag)continue;
        int Wd=Ed-1;
        for(int i=Head[x];i;i=Next[i])
            if(Val[i])Wd=Min(Wd,Dis[To[i]]);
        if(!--Cnt[Dis[x]])break;
        ++Cnt[Dis[x]=Wd+1],Cur[x]=Head[x];
        if(x!=St)x=To[Pre[x]^1];
    }
    return Flow;
}

int main()
{
    scanf("%d%d",&n,&m),St=n*m+1,Ed=St+1;
    for(int i=1;i<=n;++i)
        for(int j=1,x;j<=m;++j)
        {
            scanf("%d",&x);
            if(x==1)Add(St,ID(i,j),Inf);
            else if(x==2)Add(ID(i,j),Ed,Inf);
            for(int k=0;k<4;++k)
            {
                int wx=i+nx[k],wy=j+ny[k];
                if(wx>=1&&wx<=n&&wy>=1&&wy<=m)Add(ID(i,j),ID(wx,wy),1);
            }
        }
    printf("%d\n",ISAP());
    return 0;
}