[洛谷P1402]酒店之王

题目大意:有n个人,p个房间,q道菜,每个房间只能住一个人,每道菜只能给一个人吃。每个人有一些自己喜欢的房间和一些自己喜欢的菜。给一个人住喜欢的房间和吃喜欢的菜才能让他满意,求最多能让多少人满意。

解题思路:最大流,建立超级源点,向每个房间连容量为1的边,超级汇点,每道菜向超级汇点连接一条容量为1的边,然后每个房间向喜欢这个房间的人连接一条容量为1的边,每个人向自己喜欢的菜连接一条容量为1的边。然后你就WA了(๑ŐдŐ)b

因为这样做的话,一个人就可能住多个房间,吃多道菜,所以需要拆点。拆完发现AC了(^o^)/~

以下是Dinic算法代码。

C++ Code:

#include<cstdio>
#include<vector>
#include<cstring>
#include<queue>
using namespace std;
struct edge{
	int from,to,cap,rev;
};
vector<edge>G[220000];
queue<int>Q;
int n,p,q,iter[220000],level[220000];
inline void addedge(int from,int to,int cap){
	G[from].push_back((edge){from,to,cap,G[to].size()});
	G[to].push_back((edge){to,from,0,G[from].size()-1});
}
void bfs(int s,int t){
	memset(level,-1,sizeof level);
	level[s]=0;
	Q.push(s);
	while(!Q.empty()){
		int u=Q.front();
		Q.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;
				Q.push(e.to);
			}
		}
	}
}
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[e.to]>level[u]){
			int d=dfs(e.to,t,(e.cap>f)?(f):(e.cap));
			if(d){
				e.cap-=d;
				G[e.to][e.rev].cap+=d;
				return d;
			}
		}
	}
	return 0;
}
int max$flow(int s,int t){
	int flow=0;
	for(;;){
		bfs(s,t);
		if(level[t]<0)return flow;
		memset(iter,0,sizeof iter);
		int f;
		while(f=dfs(s,t,0x3f3f3f3f))flow+=f;
	}
}
int main(){
	scanf("%d%d%d",&n,&p,&q);
	for(int i=1;i<=n;++i)
	for(int j=1;j<=p;++j){
		int x;
		scanf("%d",&x);
		if(x){
			addedge(n*2+j,i,1);
		}
	}
	for(int i=1;i<=n;++i)
	for(int j=1;j<=q;++j){
		int x;
		scanf("%d",&x);
		if(x){
			addedge(i+n,n*2+p+j,1);
		}
	}
	for(int i=1;i<=n;++i)addedge(i,i+n,1);
	for(int i=1;i<=p;++i)addedge(0,n*2+i,1);
	for(int i=1;i<=q;++i)addedge(n*2+p+i,n*2+p+q+1,1);
	int ans=max$flow(0,n*2+p+q+1);
	printf("%d
",ans);
	return 0;
}
原文地址:https://www.cnblogs.com/Mrsrz/p/7400151.html