POJ 3281Dining(网络流之最大流）

题目地址：http://poj.org/problem?id=3281

这题WA了一晚上。。最后有幸找到了测试数据。。才发现是数组开小了。。。

不过这题的建图方法打死你我都想不到（作为刚开始网络流的菜鸟。。），建图方法是把牛放在中间，并且拆开成两份，然后左边放食物，右边放饮料。之所以把牛拆开成来两份，是为了防止一头牛占有多份食物。。（当时自己思考建图的时候就纠结着这点。。原来是这么解决的。。）代码如下：

#include <iostream>
#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <math.h>
#include <ctype.h>
#include <queue>
#include <map>
#include <algorithm>

using namespace std;
int head[2000], cnt, n, s, t, nv, maxint=9999999;
int num[2000], d[2000], pre[2000], cur[2000], q[1000000];
struct node
{
    int u, v, cap;
    int next;
}edge[1000000];
void add(int u, int v, int cap)
{
    edge[cnt].v=v;
    edge[cnt].cap=cap;
    edge[cnt].next=head[u];
    head[u]=cnt++;

    edge[cnt].v=u;
    edge[cnt].cap=0;
    edge[cnt].next=head[v];
    head[v]=cnt++;
}
void bfs()
{
    memset(num,0,sizeof(num));
    memset(d,-1,sizeof(d));
    int f1=0, f2=0, i;
    q[f1++]=t;
    num[0]=1;
    d[t]=0;
    while(f1>=f2)
    {
        int u=q[f2++];
        for(i=head[u];i!=-1;i=edge[i].next)
        {
            int v=edge[i].v;
            if(d[v]!=-1) continue ;
            d[v]=d[u]+1;
            num[d[v]]++;
            q[f1++]=v;
        }
    }
}
void isap()
{
    memcpy(cur,head,sizeof(cur));
    bfs();
    int flow=0, u=pre[s]=s, i;
    while(d[s]<nv)
    {
        //printf("%d %d %d %d\n",d[s],nv, u, t);
        if(u==t)
        {
            int f=maxint, pos;
            for(i=s;i!=t;i=edge[cur[i]].v)
            {
                if(f>edge[cur[i]].cap)
                {
                    f=edge[cur[i]].cap;
                    pos=i;
                }
            }
            for(i=s;i!=t;i=edge[cur[i]].v)
            {
                edge[cur[i]].cap-=f;
                edge[cur[i]^1].cap+=f;
            }
            flow+=f;
            u=pos;
        }
        for(i=cur[u];i!=-1;i=edge[i].next)
        {
            if(d[edge[i].v]+1==d[u]&&edge[i].cap)
            {
                break;
            }
        }
        if(i!=-1)
        {
            cur[u]=i;
            pre[edge[i].v]=u;
            u=edge[i].v;
        }
        else
        {
            if(--num[d[u]]==0) break;
            int mind=nv;
            for(i=head[u];i!=-1;i=edge[i].next)
            {
                if(mind>d[edge[i].v]&&edge[i].cap)
                {
                    cur[u]=i;
                    mind=d[edge[i].v];
                }
            }
            d[u]=mind+1;
            num[d[u]]++;
            u=pre[u];
        }
    }
    printf("%d\n",flow);
}
int main()
{
    int f, d, m, i, j, c, a, b;
    scanf("%d%d%d",&m,&f,&d);
        memset(head,-1,sizeof(head));
        cnt=0;
        s=0;
        t=2*m+f+d+1;
        nv=t+1;
        for(i=1;i<=f;i++)
        {
            add(0,i,1);
        }
        for(i=1;i<=d;i++)
        {
            add(i+2*m+f,t,1);
        }
        for(i=1;i<=m;i++)
        {
            add(f+i,f+m+i,1);
        }
        for(i=1;i<=m;i++)
        {
            scanf("%d%d",&a,&b);
            while(a--)
            {
                scanf("%d",&c);
                add(c,i+f,1);
            }
            while(b--)
            {
                scanf("%d",&c);
                add(i+f+m,c+f+2*m,1);
            }
        }
        isap();

    return 0;
}