题目描述
XX酒店的老板想成为酒店之王,本着这种希望,第一步要将酒店变得人性化。由于很多来住店的旅客有自己喜好的房间色调、阳光等,也有自己所爱的菜,但是该酒店只有p间房间,一天只有固定的q道不同的菜。
有一天来了n个客人,每个客人说出了自己喜欢哪些房间,喜欢哪道菜。但是很不幸,可能做不到让所有顾客满意(满意的条件是住进喜欢的房间,吃到喜欢的菜)。
这里要怎么分配,能使最多顾客满意呢?
输入格式:
第一行给出三个正整数表示n,p,q(<=100)。
之后n行,每行p个数包含0或1,第i个数表示喜不喜欢第i个房间(1表示喜欢,0表示不喜欢)。
之后n行,每行q个数,表示喜不喜欢第i道菜。
输出格式:
最大的顾客满意数。
这道题首先看着像二分图匹配,求最大匹配。然后我就写了。。后来思考一下不对。因为我算的不是人喜欢多少房子还是多少菜。如果按照一个人同时喜欢房子和菜建边也不太对,这样相当于
源点 ->房->人->菜->汇点的最大流量。人是可以被多次经过的,即可以有多条流入人的边,然后人多条流出。我们可以把人拆开两个点,用一条边限制经过一个人的流量。这样算出的就是每个人匹配一套的最大流量。
#include<bits/stdc++.h>
using namespace std;
const int INF=1e8+8;
const int MAXN=2e4;
struct edge{
int to,next,w;
}e[MAXN<<3];
int head[MAXN],cur[MAXN],cnt=1;
inline void add(int u,int v,int w){e[++cnt]=(edge){v,head[u],w},head[u]=cnt;}
int dep[MAXN];
queue<int>q;
int n,p,qq;
bool bfs(int x){
memset(dep,0,sizeof(dep));
q.push(x);dep[x]=1;
while(q.size()){
int u=q.front();q.pop();
for(int i=head[u];i;i=e[i].next){
int v=e[i].to,w=e[i].w;
if(w!=0&&!dep[v]){
dep[v]=dep[u]+1;q.push(v);
}
}
}
if(!dep[p+n+n+qq+1])return 0;
return 1;
}
int dfs(int u,int flow){
if(u==p+qq+n+n+1)return flow;
for(int &i=cur[u];i;i=e[i].next){
int v=e[i].to,w=e[i].w;
if(dep[v]==dep[u]+1&&w>0){
int tem=dfs(v,min(flow,w));
if(tem>0){
e[i].w-=tem;
e[i^1].w+=tem;
return tem;
}
}
}
return 0;
}
int room,meal;
int main(){
scanf("%d%d%d",&n,&p,&qq);
for(int i=1;i<=p;i++){//源点 房
add(0,i,1);add(i,0,0);
}
for(int i=1;i<=qq;i++){//汇点 菜
add(p+n+n+i,p+n+n+qq+1,1);add(p+n+n+qq+1,p+n+n+i,0);
}
for(int i=1;i<=n;i++)
for(int j=1;j<=p;j++){//房和人 j是房 p是总共
scanf("%d",&room);
if(room)add(j,p+i,1),add(p+i,j,0);
}
for(int i=1;i<=n;i++) add(p+i,p+n+i,1),add(p+n+i,p+i,0);//一个人拆成两个点,因为一个人不能住多个房子吃多组菜,限制流量为1.
for(int i=1;i<=n;i++)
for(int j=1;j<=qq;j++){//人和菜 p房总 +i是人
scanf("%d",&meal);
if(meal)add(p+n+i,p+n+n+j,1),add(p+n+n+j,p+n+i,0);
}
int ans=0;
while(bfs(0)){
for(int i=0;i<=p+n+n+qq+1;i++)cur[i]=head[i];
while(int d=dfs(0,INF))ans+=d;
}
printf("%d\n",ans);
return 0;
}