BZOJ1711 [Usaco2007 Open]Dining吃饭 题解&代码

题意:
有N头牛,F种食物和D种饮料,每头牛有多种喜欢的食物和饮料,每头牛只可以吃一种食物和饮料,且每种食物和饮料都只能被一头牛吃掉。一头牛满意当且仅当它吃到满意的食物并且喝到想喝的饮料,问最多可能让多少头牛满意。
题解:
把每头牛拆成两个点x和x+n,给x和x+n连一条容量为1的边
如果一头牛x喜欢一种食物,那么给x和食物编号点连一条容量为1的边
如果一头牛x喜欢一种饮料,那么给x+n和饮料编号点连一条容量为1的边
食物编号点和点S之间分别连一条容量为1的边
饮料编号点和点T之间分别连一条容量为1的边
然后dinic就过啦!

对了我WA了好多次后来发现自己少读了一个数据…madezz我是怎么过样例的

/**************************************************************
    Problem: 1711
    User: Rainbow6174
    Language: C++
    Result: Accepted
    Time:32 ms
    Memory:2496 kb
****************************************************************/

#include 
#include
#include 
using namespace std;
const int inf = 0x3f3f3f3f;
int n,m,mf,md,head[1005],cur[1005],cnt,st,ed,deep[1005],q[10005];
struct edge{
    int v,w,next;
} e[100005];
void add(int u, int v, int w,int rw)
{
    e[cnt]=(edge){v,w,head[u]};
    head[u]=cnt++;
    e[cnt]=(edge){u,rw,head[v]};
    head[v]=cnt++;
}
bool bfs(void)
{
    for(int i = st; i <= ed; i++)
        deep[i]=-1;
    int h=0,t=0;
    deep[st]=1;
    q[t++]=st;
    while(h != t)
    {
        int u = q[h++];
        for(int i = head[u]; i!=-1; i = e[i].next)
            if(e[i].w && deep[e[i].v]==-1)
            {
                deep[e[i].v]=deep[u]+1;
                if(e[i].v==ed) return true;
                q[t++]=e[i].v;
            }
    }
    return false;
}
int dfs(int x,int flow)
{
    if(x==ed)return flow;
    int left=flow;
    for(int i = cur[x]; i!=-1; i = e[i].next)
        if(e[i].w && deep[e[i].v]==deep[x]+1)
        {
            int tmp=dfs(e[i].v,min(left,e[i].w));
            left-=tmp;
            e[i].w-=tmp;
            e[i^1].w+=tmp;
            if(e[i].w)cur[x]=i;
            if(!left)return flow;
        }
    if(left==flow)deep[x]=-1;
    return flow-left;
}
int dinic(void)
{
    int ret=0;
    while(bfs())
    {
        for(int i = st; i <= ed; i++)
            cur[i]=head[i];
        ret+=dfs(st,inf);
    }
    return ret;
}
int main(void)
{
    scanf("%d%d%d",&n,&mf,&md);
    st=0;ed=n*2+mf+md+1;
    for(int i = st; i <= ed; i++)
        head[i]=-1;
    for(int i = 1; i <= mf; i++)
        add(st,i+n*2,1,0);
    for(int i = n*2+mf+1; i < ed; i++)
        add(i,ed,1,0);
    for(int i = 1; i <= n; i++)
        add(i,i+n,1,0);
    int t1,t2,v;
    for(int i = 1; i <= n; i++)
    {
        //printf("%d\n",i);
        scanf("%d%d",&t1,&t2);
        //printf("%d %d\n",t1,t2);
        while(t1)
        {
            scanf("%d",&v);
            add(v+n*2,i,1,0);
            t1--;
        }
        while(t2)
        {
            scanf("%d",&v);
            add(i+n,v+n*2+mf,1,0);
            t2--;
        }
    }
    printf("%d\n",dinic());
    return 0;
}

你可能感兴趣的:(网络流,BZOJ学习记录)