bzoj1412 [ZJOI2009]狼和羊的故事

题目

我们也可以很容易看出，我们就要把狼和羊分成两个集合，这也是最小割。
我们来建图。

首先，源点连向狼，容量为无限大，这样是为了防止这样被割掉。羊连向汇点，容量也为无限大。之后就是狼羊草之间的建边了。我们把狼连向草和羊，草连向草和羊，以上边容量为1。

#include<bits/stdc++.h>
#define inf ~0u>>1
#define N 10005
#define id(i,j) (i-1)*m+j
using namespace std;
int n,m,rom[105][105],x,y;
int first[N+1],siz,S,T,dis[N+1];
const int fx[4]={0,0,1,-1};
const int fy[4]={1,-1,0,0};
vector <int> nex,to,fl;
inline char nc()
{
    static char buf[100000],*p1=buf,*p2=buf;
    return p1==p2&&(p2=(p1=buf)+fread(buf,1,100000,stdin),p1==p2)?EOF:*p1++;
}
inline int read()
{
    int x=0,b=1;
    char c=nc();
    for(;!(c<='9'&&c>='0');c=nc())if(c=='-')b=-1;
    for(;c<='9'&&c>='0';c=nc())x=x*10+c-'0';
    return x*b;
}
inline void add(int x,int y,int z)
{
    nex.push_back(first[x]);
    first[x]=siz;
    to.push_back(y);
    fl.push_back(z);
    siz++;
}
inline void insert(int x,int y,int z)
{
    add(x,y,z),add(y,x,0);
}
inline bool BFS(int s,int t)
{
    queue <int> Q;
    for(int i=S;i<=T;i++)dis[i]=inf;
    dis[s]=0,Q.push(s);
    while(!Q.empty())
    {
        int x=Q.front();Q.pop();
        for(int i=first[x];i!=-1;i=nex[i])
        {
            int u=to[i];
            if(fl[i]&&dis[x]+1<dis[u])
            {
                dis[u]=dis[x]+1;
                if(u==t)return true;
                Q.push(u);
            }
        }
    }
    if(dis[t]==inf)return false;return true;
}
inline int DFS(int s,int f,int t)
{
    if(s==t)return f;
    int rest=f;
    for(int i=first[s];i!=-1;i=nex[i])
    {
        int u=to[i];
        if(rest&&fl[i]&&dis[u]==dis[s]+1)
        {
            int tmp=DFS(u,min(rest,fl[i]),t);
            if(tmp==0)dis[u]=0;
            fl[i]-=tmp;
            fl[i^1]+=tmp;
            rest-=tmp;
        }
    }
    return f-rest;
}
inline int Dinic(int s,int t)
{
    int ans=0;
    while(BFS(s,t))ans+=DFS(s,inf,t);
    return ans;
}
inline bool in(int x,int y)
{
    return x<=n&&x>=1&&y<=m&&y>=1;
}
int main()
{
    //freopen("in.txt","r",stdin);
    memset(first,-1,sizeof(first));
    n=read(),m=read();
    S=0,T=m*n+1;
    for(int i=1;i<=n;i++)
        for(int j=1;j<=m;j++)
            rom[i][j]=read();
    for(int i=1;i<=n;i++)
    {
        for(int j=1;j<=m;j++)
        {
            if(rom[i][j]==1)insert(S,id(i,j),inf);
            if(rom[i][j]==2)insert(id(i,j),T,inf);
            for(int k=0;k<=3;k++)
            {
                x=i+fx[k],y=j+fy[k];
                if(!in(x,y))continue;
                if(rom[i][j]==1&&(rom[x][y]==0||rom[x][y]==2))insert(id(i,j),id(x,y),1);
                if(rom[i][j]==0&&(rom[x][y]==0||rom[x][y]==2))insert(id(i,j),id(x,y),1);
            }
        }
    }
    cout<<Dinic(S,T);
    return 0;
}