[BZOJ1711]Dining 吃饭 最大流

核心思想:用一个流量表示一个方案。
源点向每个食物连一条容量为1的边。
每个饮料向汇点连一条容量为1的边。
把每个奶牛拆成两个点,入点向出点连一条容量为1的边。
每个食物向能吃它的奶牛的入点连一条容量为1的边。
每个奶牛的出点向它能喝的饮料连一条容量为1的边。
1流量代表一个奶牛吃饱,跑最大流即可。
代码:

#include
#include
#include 
using namespace std;
int n,r,d,S,T,dl[410],ne[410];
bool vis[410];
struct edge
{
    int t,c,f;
    edge *next,*rev;
}*con[410];
void ins(int x,int y,int c)
{
    edge *p=new edge;p->t=y;p->c=c;p->f=0;p->next=con[x];con[x]=p;
    p=new edge;p->t=x;p->c=0;p->f=0;p->next=con[y];con[y]=p;
    con[x]->rev=con[y];con[y]->rev=con[x];
}
bool bfs()
{
    bool re=0;
    memset(ne,0,sizeof(ne));
    ne[S]=1;dl[1]=S;
    for(int head=1,tail=1;head<=tail;head++)
    {
        int v=dl[head];
        if(v==T) re=1;
        for(edge *p=con[v];p;p=p->next)
            if(p->c>p->f&&ne[p->t]==0)
                ne[p->t]=ne[v]+1,dl[++tail]=p->t;
    }
    return re;
}
int dinic(int v,int flow)
{
    if(v==T) return flow;
    if(vis[v]) return 0;
    int re=0;
    for(edge *p=con[v];p;p=p->next)
        if(p->c>p->f&&ne[p->t]==ne[v]+1)
        {
            int tmp=dinic(p->t,min(flow,p->c-p->f));
            re+=tmp;p->f+=tmp;
            flow-=tmp;p->rev->f-=tmp;
            if(flow==0) break;
        }
    if(re==0) vis[v]=1;
    return re;  
}
int main()
{

    scanf("%d%d%d",&n,&r,&d);
    S=0;T=2*n+r+d+1;
    for(int i=1;i<=r;i++)
        ins(S,i,1);
    for(int i=1;i<=d;i++)
        ins(i+2*n+r,T,1);
    for(int i=1;i<=n;i++)
        ins(r+2*i-1,r+2*i,1);   
    for(int i=1;i<=n;i++)
    {
        int nr,nd,x;
        scanf("%d%d",&nr,&nd);
        for(int j=1;j<=nr;j++)
        {
            scanf("%d",&x);
            ins(x,r+2*i-1,1);
        }
        for(int j=1;j<=nd;j++)
        {
            scanf("%d",&x);
            ins(r+2*i,2*n+r+x,1);
        }
    }
    int ans=0;
    while(bfs())
    {
        memset(vis,0,sizeof(vis));
        ans+=dinic(S,0x3f3f3f3f);
    }
    printf("%d",ans);
    return 0;
}

你可能感兴趣的:(网络流)