POJ - 3281 (Dining)

题意：有 N 头牛，F种食物和D种饮料，每头牛都有喜欢的食物集合和饮料集合，如果在两个集合中分别任选一个就可以让这头牛满意，现给出N头牛的喜爱的食物和饮料集合，求F种食物和D种饮料每个只有一份的情况下最多能满足牛的数量；

 

分析：网络流建图，设 0 为源点，1~F 为食物的种类编号，F+1~F+D为饮料的种类编号，F+D+2*i-1~F+D+2*i 为第i头牛的编号(后面解释为什么一头牛需要两个点)，F+D+2*N+1 为汇点，则根据题意：

① 0 和 1~F 分别建一条容量为 1 的边；

② F+1~F+D 分别 和 F+D+2*N+1 建一条容量为 1 的边；

③ 若1~F 中点 j 满足是第 i 头牛喜爱的食物的话则，由 j 向 F+D+2*i-1 连一条容量为 1 的边；

④ 若F+1~F+D 中点 j 满足是第 i 头牛喜爱的饮料的话则，由 F+D+2*i 向 j 连一条容量为 1 的边；

⑤ F+D+2*i-1 向 F+D+2*i 连一条容量为 1 的边；

以上建图后，由源点到汇点的一条路径即表示满足一头牛的情况，解释一下⑤，因为每头牛只要满足要求的一份食物和饮料就行，所以得把每头牛拆成两个点中间连一条容量为 1 的边，这样才能限制只有至多一条路径经过一头牛；

再套个求最大流的板子就好了，这里我用的是 Dinic

 

代码：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<queue>
#define N 1010
#define M 10010
#define INF 0x7fffffff
using namespace std;
int n,F,D,ss,tt;
int dis[N];
queue<int> q;

struct Edge{int to;int value;int next;}e[M<<1];
int head[N],cnt=-1;
void add(int from,int to,int value)
{
    cnt++;
    e[cnt].to=to;
    e[cnt].value=value;
    e[cnt].next=head[from];
    head[from]=cnt;
}

bool bfs(int s,int t)
{
    q=queue<int>();
    memset(dis,-1,sizeof(dis));
    dis[s]=0;
    q.push(s);
    while(!q.empty())
    {
        int x=q.front();
        q.pop();
        for(int i=head[x];i>-1;i=e[i].next)
        {
            int now=e[i].to;
            if(dis[now]==-1&&e[i].value!=0)
            {
                dis[now]=dis[x]+1;
                q.push(now);
            }
        }
    }
    return dis[t]!=-1;
}
int cur[N];

int dfs(int x,int t,int maxflow)
{
    if(x==t)return maxflow;
    int ans=0;
    for(int i=cur[x];i>-1;i=e[i].next)
    {
        int now=e[i].to;
        if(dis[now]!=dis[x]+1||e[i].value==0||ans>=maxflow)continue;
        cur[x]=i;
        int f=dfs(now,t,min(e[i].value,maxflow-ans));
        e[i].value-=f;
        e[i^1].value+=f;
        ans+=f;
    }
    return ans;
}

int Dinic(int s,int t)
{
    int ans=0;
    while(bfs(s,t))
    {
        memcpy(cur,head,sizeof(head));
        ans+=dfs(s,t,INF);
    }
    return ans;
}

int main()
{
    memset(head,-1,sizeof(head));
    scanf("%d%d%d",&n,&F,&D);
    for(int i=1;i<=n;i++)
    {
    	int f,d,x; scanf("%d%d",&f,&d);
    	while(f--){
    		scanf("%d",&x);
			add(x,F+D+2*i-1,1);
			add(F+D+2*i-1,x,0); 
		}
		while(d--){
			scanf("%d",&x);
			add(F+D+2*i,F+x,1);
			add(F+x,F+D+2*i,0); 
		}
		add(F+D+2*i-1,F+D+2*i,1);
		add(F+D+2*i,F+D+2*i-1,0);
    }
    for(int i=1;i<=F;i++){  //Start->F
    	add(0,i,1);
    	add(i,0,0);
	}
	for(int i=F+1;i<=F+D;i++){  //D->end
		add(i,F+D+2*n+1,1);
		add(F+D+2*n+1,i,0);
	}
	n=F+D+2*n+1;
    printf("%d\n",Dinic(0,n));
    return 0;
}