Dining POJ - 3281 Dining 拆点+最大流

传送门：POJ3281

题意：有N头牛，F种食物和D种饮料，每头牛都有各自喜欢的食物和饮料，每种饮料或食物只能分配给一头牛，问最多有多少头牛能同时得到自己喜欢的食物和饮料。

思路：一开始把饮料和食物放在一起建图， 想了很久也没想出个所以然来。

正解：将饮料和食物分别放在两边，中间用牛隔开，并且牛要拆点，这样才能限制每头牛只能选一个食物和饮料（即限制点流量）。

详细建图解释参考挑战程序设计P234.

代码：

//ISAP int
#include <iostream>
#include <cstdio>
#include <cstring>
#include <queue>
#define ll long long
#define MAXN 505
#define inf 0x3f3f3f3f
using namespace std;
int n,m;//题目输入点数，边数
struct Edge{
	int v,next;
	int cap,flow;
}edge[MAXN*1000];
int cur[MAXN],pre[MAXN],gap[MAXN],path[MAXN],dep[MAXN];
int cnt=0;//实际存储总边数 
void init()
{
	cnt=0;
	memset(pre,-1,sizeof(pre)); 
}
void add(int u,int v,int w,int rw=0)//加边 有向图三个参数，无向图四个 
{
	edge[cnt].v=v;
	edge[cnt].cap=w;
	edge[cnt].flow=0;
	edge[cnt].next=pre[u];
	pre[u]=cnt++;
	edge[cnt].v=u;
	edge[cnt].cap=rw;
	edge[cnt].flow=0;
	edge[cnt].next=pre[v];
	pre[v]=cnt++;
}
bool bfs(int s,int t)//其实这个bfs可以融合到下面的迭代里，但是好像是时间要长 
{
	memset(dep,-1,sizeof(dep));
	memset(gap,0,sizeof(gap));
	gap[0]=1;
	dep[t]=0;
	queue<int>q;
	while(!q.empty())
	q.pop();
	q.push(t);//从汇点开始反向建层次图 
	while(!q.empty())
	{
		int u=q.front();
		q.pop();
		for(int i=pre[u];i!=-1;i=edge[i].next)
		{
			int v=edge[i].v;
			if(dep[v]==-1&&edge[i^1].cap>edge[i^1].flow)//注意是从汇点反向bfs，但应该判断正向弧的余量
			{
				dep[v]=dep[u]+1;
				gap[dep[v]]++;
				q.push(v);
				//if(v==s)//感觉这两句优化加了一般没错，但是有的题可能会错，所以还是注释出来，到时候视情况而定 
				//break;
			}	
		}
	}
	return dep[s]!=-1; 
}
int isap(int s,int t)
{
	bfs(s,t);
	memcpy(cur,pre,sizeof(pre));
	int u=s;
	path[u]=-1;
	int ans=0;
	while(dep[s]<n)//迭代寻找增广路 
	{
		if(u==t)
		{
			int f=inf;
			for(int i=path[u];i!=-1;i=path[edge[i^1].v])//修改找到的增广路 
				f=min(f,edge[i].cap-edge[i].flow);
			for(int i=path[u];i!=-1;i=path[edge[i^1].v])
			{
				edge[i].flow+=f;
				edge[i^1].flow-=f;
			}
			ans+=f;
			u=s;
			continue;
		}
		bool flag=false;
		int v;
		for(int i=cur[u];i!=-1;i=edge[i].next)
		{
			v=edge[i].v;
			if(dep[v]+1==dep[u]&&edge[i].cap-edge[i].flow)
			{
				cur[u]=path[v]=i;//当前弧优化 
				flag=true;
				break;
			}
		}
		if(flag)
		{
			u=v;
			continue;
		}
		int x=n;
		if(!(--gap[dep[u]]))return ans;//gap优化 
		for(int i=pre[u];i!=-1;i=edge[i].next)
		{
			if(edge[i].cap-edge[i].flow&&dep[edge[i].v]<x)
			{
				x=dep[edge[i].v];
				cur[u]=i;//常数优化 
			}
		}
		dep[u]=x+1;
		gap[dep[u]]++;
		if(u!=s)//当前点没有增广路则后退一个点 
		u=edge[path[u]^1].v;
	 } 
	 return ans;
}
int main()
{
	int N,F,D,a,b,t;
	init();
	cin>>N>>F>>D;
	int source=0,sink=n=N*2+F+D+1;
	for(int i=1;i<=F;i++)
	add(source,i,1);
	for(int i=1;i<=D;i++)
	add(i+N*2+F,sink,1);
	for(int i=1;i<=N;i++)
	{
		add(i+F,i+F+N,1);
		scanf("%d%d",&a,&b);
		while(a--)
		{
			scanf("%d",&t);
			add(t,i+F,1);
		 }
		while(b--)
		{
			scanf("%d",&t);
			add(i+F+N,t+N*2+F,1);
		}
	}
	cout<<isap(source,sink)<<endl;
 }