poj3281 Dining Dinic

由于牛只能选一次，所以对牛进行拆点。

#include <iostream>
#include <algorithm>
#include <cstring>
#include <cstdio>
#include <queue>
#include <cmath>
using namespace std;
#define maxn 400000
#define INF 0xfffffff
struct node
{
    int v,c,next;
}e[maxn];
int st,ed,en;
int q[maxn];
int first[maxn],dis[maxn];
int n,m,f,d;
void init()
{
    en=0;
    st=f+n*2+d+2;
    ed=f+n*2+d+1;
    memset(first,-1,sizeof(first));
}
void add(int a,int b,int c)
{
    e[en].c=c;
    e[en].v=b;
    e[en].next=first[a];
    first[a]=en;
    en++;
    e[en].c=0;
    e[en].v=a;
    e[en].next=first[b];
    first[b]=en;
    en++;
}
bool bfs()
{
    int head=1;
    int tail=1;
    memset(dis,-1,sizeof(dis));
    dis[st]=0;
    tail++;
    q[tail]=st;
    int u,v;
    while(head<tail)
    {
        head++;
        u=q[head];
        for(int i=first[u];i!=-1;i=e[i].next)
        {
            v=e[i].v;
            if(dis[v]==-1&&e[i].c)
            {
                dis[v]=dis[u]+1;
                tail++;
                q[tail]=v;
                if(v==ed)
                {
                    return true;
                }
            }
        }
    }
    return false;
}
int dfs(int x,int mx)
{
    if(x==ed||mx==0) return mx;
    int flow=0,v,f;
    for(int i=first[x];i!=-1;i=e[i].next)
    {
        v=e[i].v;
        if( dis[v]==dis[x]+1 && (f=dfs(v,min(mx,e[i].c))) )
        {
            e[i].c-=f;
            e[i^1].c+=f;
            flow+=f;
            mx-=f;
            if(!mx) break;
        }
    }
    return flow;
}
int dinic()
{
    int tmp=0,maxflow=0;
    while(bfs())
    {
        //printf("in\n");
        while(tmp=dfs(st,INF))
        {
            //printf("%d\n",tmp);
            //system("pause");
            maxflow+=tmp;
        }
    }
    return maxflow;
}
void build()
{
    int a,b,c;
    for(int i=1;i<=n;i++)
    {
        add(f+i,f+i+n,1);
    }
    for(int i=1;i<=f;i++)
    {
        add(st,i,1);
    }
    for(int i=1;i<=d;i++)
    {
        add(n*2+f+i,ed,1);
    }
    for(int i=1;i<=n;i++)
    {
        scanf("%d%d",&a,&b);
        for(int j=1;j<=a;j++)
        {
            scanf("%d",&c);
            add(c,f+i,1);
        }
        for(int j=1;j<=b;j++)
        {
            scanf("%d",&c);
            add(f+i+n,2*n+f+c,1);
        }
    }
}
int main()
{
    while(~scanf("%d%d%d",&n,&f,&d))
    {
        init();
        build();
        //printf("orz\n");
        printf("%d\n",dinic());
    }
    return 0;
}