Poj-3281 Dining（拆点限流）

建模如下图：为满足每种事物或饮料只能分配给一头牛，对牛进行拆分，之间流量为1。

#include<iostream>
#include<cstdio>
#include<vector>
#include<queue>
#include<cstring>
#include<algorithm>
using namespace std;
typedef long long ll;
const int inf=0x3f3f3f3f;
const int Max_n=1e5+10;

int n,f,d;
struct edge{
	int to,cap,rev;
	edge(int t,int c,int r):to(t),cap(c),rev(r){}
}; 
vector<edge>G[Max_n];
int level[Max_n];  //顶点到源点的距离标号
int iter[Max_n];   //当前弧优化（标记之前已经用过的边）

void add_edge(int from,int to,int cap){
	G[from].push_back(edge(to,cap,G[to].size()));
	G[to].push_back(edge(from,0,G[from].size()-1));
} 

//通过BFS构造层次图
void bfs(int s){
	memset(level,-1,sizeof(level));
	queue<int>que;
	level[s]=0;
	que.push(s);
	while(!que.empty()){
		int u=que.front();que.pop();
		for(int i=0;i<G[u].size();i++){
			edge e=G[u][i];
			if(e.cap>0&&level[e.to]<0){
				level[e.to]=level[u]+1;
				que.push(e.to);
			}
		}
	}
} 

//在层次图上通过dfs不断寻找增光路
int dfs(int u,int t,int f){
	if(u==t)return f;
	for(int& i=iter[u];i<G[u].size();i++){ //当前弧优化
		edge& e=G[u][i];
		if(e.cap>0&&level[u]<level[e.to]){
			int d=dfs(e.to,t,min(e.cap,f));
			if(d>0){
				e.cap-=d;
				G[e.to][e.rev].cap+=d;
				return d;
			}
		} 
	}
	return -1;
} 

int max_flow(int s,int t){
	int flow=0;
	while(true){
		bfs(s);
		if(level[t]<0)return flow;
		memset(iter,0,sizeof(iter));
		int f;
		while((f=dfs(s,t,inf))>0){
			flow+=f;
		}
	}
}

int main()
{
	int s=0,t=404;
	int dish=0,cowa=100,cowb=200,drink=300;
	scanf("%d%d%d",&n,&f,&d);
	for(int i=1;i<=f;i++){ //s-dish 
		add_edge(s,dish+i,1);
	}
	for(int i=1;i<=n;i++){ //cowa-cowb
		add_edge(cowa+i,cowb+i,1);
	}
	for(int i=1;i<=d;i++){ //drink-t
		add_edge(drink+i,t,1);
	}
	int a,b,c;
	for(int i=1;i<=n;i++){ 
		scanf("%d%d",&a,&b);
		for(int j=0;j<a;j++){
			scanf("%d",&c);
			add_edge(dish+c,cowa+i,inf);
		}
		for(int j=0;j<b;j++){
			scanf("%d",&c);
			add_edge(cowb+i,drink+c,inf); 
		}
	}
	printf("%d\n",max_flow(s,t));
	return 0;
}