luogu P1402 酒店之王

题面传送门
一道二分图的题目。
考虑将菜向客人,客人向房间连权值为 1 1 1的边,但是这样会有重复计算。
所以将客人拆成两个点,两点之间连权值为 1 1 1的边,就可以达到限制流量的目的。
代码实现:

#include
#include
#include
#define min(a,b) ((a)<(b)?(a):(b))
using namespace std;
int n,m,k,x,y,z,st,t,nows,cur,now[1039],d[1039],ans;
struct yyy{
     int to,w,z;}tmp;
struct ljb{
     
	int head,h[1039];
	yyy f[10039];
	inline void add(int x,int y,int z){
     
		f[head]=(yyy){
     y,z,h[x]};
		h[x]=head++;
	}
}s;
queue<int > q;
inline int bfs(){
     
	while(!q.empty()) q.pop();
	memset(d,0x3f,sizeof(d));
	d[st]=0;
	q.push(st);
	now[st]=s.h[st];
	while(!q.empty()){
     
		nows=q.front();
		q.pop();
		cur=s.h[nows];
		while(cur!=-1){
     
			tmp=s.f[cur];
			if(tmp.w&&d[tmp.to]>=1e9){
     
				d[tmp.to]=d[nows]+1;q.push(tmp.to);now[tmp.to]=s.h[tmp.to];
				if(tmp.to==t) return 1;
			}
			cur=tmp.z;
		}
	}
	return 0;
}
inline int dfs(int x,int sum){
     
	if(x==t) return sum;
	int cur=s.h[x],k,pus=0;
	yyy tmp;
	while(cur!=-1){
     
		tmp=s.f[cur];
		if(tmp.w&&d[tmp.to]==d[x]+1){
     
			k=dfs(tmp.to,min(sum,tmp.w));
			if(!k) d[tmp.to]=1e9;
			s.f[cur].w-=k;
			s.f[cur^1].w+=k;
			sum-=k;
			pus+=k;
		}
		cur=tmp.z;
		if(!sum) break;
	}
	return pus;
}
int main(){
     
	memset(s.h,-1,sizeof(s.h));
	register int i,j;
	scanf("%d%d%d",&n,&m,&k);
	st=0;t=m+2*n+k+1;
	for(i=1;i<=m;i++) s.add(0,i,1),s.add(i,0,0);
	for(i=1;i<=n;i++){
     
		for(j=1;j<=m;j++){
     
			scanf("%d",&x);
			if(x) s.add(j,i+m,1),s.add(i+m,j,0);
		}
	}
	for(i=1;i<=n;i++) s.add(i+m,i+n+m,1),s.add(i+n+m,i+m,0);
	for(i=1;i<=n;i++){
     
		for(j=1;j<=k;j++){
     
			scanf("%d",&x);
			if(x) s.add(i+2*n,j+2*n+m,1),s.add(j+2*n+m,i+2*n,0);
		}
	}
	for(i=1;i<=k;i++) s.add(i+2*n+m,t,1),s.add(t,i+2*n+m,0);
	while(bfs()) ans+=dfs(st,1e8);
	printf("%d\n",ans);
}

你可能感兴趣的:(洛谷,二分图,拆点)