POJ 3281 Dining(最大流)

题目链接：http://poj.org/problem?id=3281

题意：有一些牛每个牛都有自己喜欢吃的food和喜欢喝的drink，饲养员有一些种类的food和一些种类的drink，合理分配这些食物和饮料，使得能够同时得到自己喜欢的食物和饮料（只要一个食物和一个饮料）的牛的数量最多。并且一种饮料或食物只能分配给一头牛。

题解：要求牛的数量于是，将牛拆成两个点一条边，在边上加上容量。然后设置一个起点，连接所有food，容量设置为1，设置一个终点，连接到所有drink，容易设置为1，将每个牛和自己喜欢的食物和饮料连接， 从起点到终点求最大流。

代码：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<queue>
const int MAX=400+10;
const int INF=0x3f3f3f3f;
using namespace std;
int flow[MAX][MAX],cap[MAX][MAX];
int res[MAX],pre[MAX];
int N;
int EK(int s,int e)
{
	int ans=0;
	queue<int> que;
	while(1)
	{
		memset(res,0,sizeof(res));
		que.push(s);
		res[s]=INF;
		while(!que.empty())
		{
			int u=que.front();
			que.pop();
			for(int v=0;v<N;v++)
			{
				if(!res[v]&&cap[u][v]>flow[u][v])
				{
					pre[v]=u;
					que.push(v);
					res[v]=min(res[u],cap[u][v]-flow[u][v]);
				}
			}
		}
		if(res[e]==0) break;
		ans+=res[e];
		for(int i=e;i!=s;i=pre[i])
		{
			flow[pre[i]][i]+=res[e];
			flow[i][pre[i]]-=res[e];
		}
	}
	return ans;
}
int main()
{
	//freopen("in.txt","r",stdin);
	//freopen("out.txt","w",stdout);
	int f,d,n;
	while(scanf("%d%d%d",&n,&f,&d)!=EOF)
	{
		memset(cap,0,sizeof(cap));
		memset(flow,0,sizeof(flow));
		N=f+d+n*2+2;
		int s=0;
		int e=N-1;
		for(int i=1;i<=f;i++)
			cap[s][i]=1;
		for(int i=1;i<=d;i++)
			cap[f+n*2+i][e]=1;
		for(int i=1;i<=n;i++)
		{
			cap[f+i][f+n+i]=1;
			int fnum,dnum;
			scanf("%d%d",&fnum,&dnum);
			while(fnum--)
			{
				int food;
				scanf("%d",&food);
				cap[food][f+i]=1;
			}
			while(dnum--)
			{
				int drink;
				scanf("%d",&drink);
				cap[f+n+i][f+2*n+drink]=1;
			}
		}
		printf("%d\n",EK(0,N-1));
	/*	for(int i=1;i<=n;i++)
		{
			cout<<i<<":";
			for(int j=1;j<=f;j++)
			{
				if(flow[j][f+i]==1) cout<<"food "<<j<<" ";
			}
			for(int j=1;j<=d;j++)
			{
				if(flow[f+n+i][j+f+2*n]==1) cout<<"drink "<<j<<endl;
			}
		}*/
	}
	return 0;
}