HDU 3046

http://acm.hdu.edu.cn/showproblem.php?pid=3046

典型的最小割模型

#include <iostream> 

#include <cstdio>

#include <cstring>

using namespace std ;



const int INF=0xfffffff ;

struct node

{

    int s,t,cap,nxt ;

}e[400005] ;

int m,n,cnt,head[100005],level[100005],q[100005] ;

void add(int s,int t,int cap)

{

    e[cnt].s=s ;e[cnt].t=t ;e[cnt].cap=cap ;e[cnt].nxt=head[s] ;head[s]=cnt++ ;

    e[cnt].s=t ;e[cnt].t=s ;e[cnt].cap=0 ;e[cnt].nxt=head[t] ;head[t]=cnt++ ;

}

bool build(int s,int t)

{

    int front=0,rear=0 ;

    memset(level,-1,sizeof(level)) ;

    q[rear++]=s ;

    level[s]=1 ;

    while(front<rear)

    {

        int u=q[front++] ;

        for(int i=head[u] ;i!=-1 ;i=e[i].nxt)

        {

            int tt=e[i].t ;

            if(level[tt]==-1 && e[i].cap>0)

            {

                level[tt]=level[u]+1 ;

                if(tt==t)return true ;

                q[rear++]=tt ;

            }

        }

    }

    return false ;

}

int find(int s,int t,int flow)

{

    if(s==t)return flow ;

    int ret=0,a ;

    for(int i=head[s] ;i!=-1 ;i=e[i].nxt)

    {

        int tt=e[i].t ;

        if(level[tt]==level[s]+1 && e[i].cap>0)

        {

            a=find(tt,t,min(e[i].cap,flow-ret)) ;

            e[i].cap-=a ;

            e[i^1].cap+=a ;

            ret+=a ;

            if(ret==flow)

                return ret ;

        }

    }

    if(!ret)level[s]=-1 ;

    return ret ;

}

int dinic(int s,int t)

{

    int flow,ret=0 ;

    while(build(s,t))

        while(flow=find(s,t,INF))

            ret+=flow ;

    return ret ;

}

int Map[205][205] ;

int main()

{

    int N,M ;

    int cas=1 ;

    while(~scanf("%d%d",&N,&M))

    {

        cnt=0 ;

        memset(head,-1,sizeof(head)) ;

        int S,T ; 

        for(int i=1 ;i<=N ;i++)

        {

            for(int j=1 ;j<=M ;j++)

            {

                scanf("%d",&Map[i][j]) ;

            }

        }

        S=0 ;T=N*M+1 ;

        for(int i=1 ;i<=N ;i++)

        {

            for(int j=1 ;j<=M ;j++)

            {

                int num=(i-1)*M+j ;

                if(i>1)add(num,num-M,1) ;

                if(i<N)add(num,num+M,1) ;

                if(j>1)add(num,num-1,1) ;

                if(j<M)add(num,num+1,1) ;

                if(Map[i][j]==1)add(S,num,INF) ;

                if(Map[i][j]==2)add(num,T,INF) ;

            }

        }

        printf("Case %d:\n%d\n",cas++,dinic(S,T)) ;

    }

    return 0 ;

}

View Code