Poj-3281 Dining(拆点限流)

建模如下图:为满足每种事物或饮料只能分配给一头牛,对牛进行拆分,之间流量为1。
Poj-3281 Dining(拆点限流)_第1张图片

#include
#include
#include
#include
#include
#include
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){}
}; 
vectorG[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));
	queueque;
	level[s]=0;
	que.push(s);
	while(!que.empty()){
		int u=que.front();que.pop();
		for(int i=0;i0&&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];i0&&level[u]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

你可能感兴趣的:(Acm--模板题)