首页 > 代码库 > HDU 4888

HDU 4888

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

这题时限给的太短了,以至于卡了一些奇怪的东西,我开始先输入再建图一直T,一边输入一边连边就AC了

题意:n*m的矩阵,每个点可以放0-k的数,给你每行每列的和,问可行否,可行的话解是否唯一,唯一则输出那组解

网络流行列建边,容量为k,这样一条边代表一个点,起点和每行连容量为此行和的边,列和终点连容量为每列和的边,跑最大流,看是否满流

如果能在残量网络找到环,则答案不唯一(按环走一遍不改变结果),注意找环的时候不能马上找其反向边,所以dfs要加一个参数记录当前节点的父亲

输出答案的时候只要看行列每边的流量,流量=k-剩余流量

#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 S,T ;int vis[100005] ;int JG(int u,int fa){    if(vis[u])return 1 ;    vis[u]=1 ;    for(int i=head[u] ;i!=-1 ;i=e[i].nxt)    {        int v=e[i].t ;        if(v!=fa && v!=S && v!=T && e[i].cap)            if(JG(v,u))return 1 ;    }    vis[u]=0 ;    return 0 ;}int A[505],B[505] ;int res[505][505] ;int main(){    int k ;    while(~scanf("%d%d%d",&n,&m,&k))    {        cnt=0 ;        memset(head,-1,sizeof(head)) ;        int sum1=0,sum2=0 ;        S=0 ;T=n+m+1 ;        for(int i=1 ;i<=n ;i++)        {            scanf("%d",&A[i]) ;            sum1+=A[i] ;            add(S,i,A[i]) ;            for(int j=1 ;j<=m ;j++)                add(i,j+n,k) ;        }        for(int i=1 ;i<=m ;i++)        {            scanf("%d",&B[i]) ;            sum2+=B[i] ;            add(i+n,T,B[i]) ;        }        int ans=dinic(S,T) ;        if(ans==sum1 && ans==sum2)        {            int flag=0 ;            memset(vis,0,sizeof(vis)) ;            for(int i=1 ;i<=n ;i++)            {                if(JG(i,-1))                {                    flag=1 ;                    break ;                }            }            if(flag)            {                puts("Not Unique") ;            }            else            {                puts("Unique") ;                memset(res,0,sizeof(res)) ;                for(int i=1 ;i<=n ;i++)                {                    for(int j=head[i] ;j!=-1 ;j=e[j].nxt)                    {                        if(e[j].t<=n+m && e[j].t>n)                            res[i][e[j].t-n]=k-e[j].cap ;                    }                }                for(int i=1 ;i<=n ;i++)                {                    for(int j=1 ;j<=m ;j++)                     {                         if(j>1)printf(" ") ;                         printf("%d",res[i][j]) ;                    }                    putchar(\n) ;                }            }        }        else        {            puts("Impossible") ;        }    }    return 0 ;}
View Code