AC日记——Dining poj 3281

[POJ-3281]

思路：

　　把牛拆点；

　　s向食物连边，流量1；

　　饮料向t连边，流量1；

　　食物向牛1连边，流量1；

　　牛2向饮料连边，流量1；

　　最大流；

来，上代码：

#include <cstdio>

#include <cstring>

#include <iostream>

#include <algorithm>

using namespace std;

#define maxn 2005

#define INF 0x7fffffff

int n,f,d,s,t,head[maxn],E[maxn<<],V[maxn<<],F[maxn<<];

int cnt=,deep[maxn],que[maxn<<];

inline void in(int &now)

{

    char Cget=getchar();now=;

    while(Cget>''||Cget<'') Cget=getchar();

    while(Cget>=''&&Cget<='')

    {

        now=now*+Cget-'';

        Cget=getchar();

    }

}

inline void edge_add(int u,int v,int f)

{

    E[++cnt]=head[u],V[cnt]=v,F[cnt]=f,head[u]=cnt;

    E[++cnt]=head[v],V[cnt]=u,F[cnt]=,head[v]=cnt;

}

inline bool bfs()

{

    for(int i=s;i<=t;i++) deep[i]=-;

    int h=,tail=;deep[s]=,que[h]=s;

    while(h<tail)

    {

        int now=que[h++];

        for(int i=head[now];i;i=E[i])

        {

            if(deep[V[i]]<&&F[i]>)

            {

                deep[V[i]]=deep[now]+;

                if(V[i]==t) return true;

                que[tail++]=V[i];

            }

        }

    }

    return false;

}

int flowing(int now,int flow)

{

    if(flow<=||now==t) return flow;

    int oldflow=;

    for(int i=head[now];i;i=E[i])

    {

        if(F[i]>&&deep[V[i]]==deep[now]+)

        {

            int pos=flowing(V[i],min(flow,F[i]));

            flow-=pos,oldflow+=pos,F[i]-=pos,F[i^]+=pos;

            if(flow==) return oldflow;

        }

    }

    if(oldflow==) deep[now]=-;

    return oldflow;

}

int main()

{

    in(n),in(f),in(d);

    int u,v,nf,nd;t=f+n+n+d+;

    for(int i=;i<=f;i++) edge_add(s,i,);

    for(int i=;i<=d;i++) edge_add(f+n+n+i,t,);

    for(int i=;i<=n;i++)

    {

        edge_add(f+i,f+n+i,),in(nf),in(nd);

        for(int j=;j<=nf;j++) in(u),edge_add(u,f+i,);

        for(int j=;j<=nd;j++) in(u),edge_add(f+n+i,f+n+n+u,);

    }

    int ans=;

    while(bfs()) ans+=flowing(s,INF);

    cout<<ans;

    return ;

}