首页 > 代码库 > HDU 3046

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