洛谷P1402 酒店之王【多部图匹配】

时空限制 1000ms / 128MB

题目描述

XX酒店的老板想成为酒店之王，本着这种希望，第一步要将酒店变得人性化。由于很多来住店的旅客有自己喜好的房间色调、阳光等，也有自己所爱的菜，但是该酒店只有p间房间，一天只有固定的q道不同的菜。

有一天来了n个客人，每个客人说出了自己喜欢哪些房间，喜欢哪道菜。但是很不幸，可能做不到让所有顾客满意（满意的条件是住进喜欢的房间，吃到喜欢的菜）。
这里要怎么分配，能使最多顾客满意呢？

输入格式：

第一行给出三个正整数表示n,p,q(<=100)。

之后n行，每行p个数包含0或1，第i个数表示喜不喜欢第i个房间（1表示喜欢，0表示不喜欢）。

之后n行，每行q个数，表示喜不喜欢第i道菜。

输出格式：

最大的顾客满意数。

---

题目分析

这个题目一眼就看得出是匹配
但是点集有三个，怎么办呢
以这题为例

我们把n个客人拆成2*n个点
i与i+n连一条连一条容量为1的边
保证了每个客人只被匹配一次

每个客人由自己喜欢的房间连边 ，容量为1
每个客人向喜欢的菜连边，容量为1

建立超级汇点与超级源点
超级源点向每个房间连边，容量为1
每个房间向超级菜连边，容量为1
然后跑最大流即可

---

#include<iostream>
#include<cmath>
#include<algorithm>
#include<queue>
#include<cstring>
#include<cstdio>
using namespace std;

int read()
{
    int f=1,x=0;
    char ss=getchar();
    while(ss<'0'||ss>'9'){if(ss=='-')f=-1;ss=getchar();}
    while(ss>='0'&&ss<='9'){x=x*10+ss-'0';ss=getchar();}
    return f*x;
}

const int inf=1e9;
const int maxn=1010;
int n,p,q,s=0,t;
struct node{int v,f,nxt;}E[maxn*200];
int head[maxn],tot=1;
int lev[maxn],maxf;

void add(int u,int v,int f)
{
    E[++tot].nxt=head[u];
    E[tot].v=v; E[tot].f=f;
    head[u]=tot;
}

int bfs()
{
    memset(lev,-1,sizeof(lev)); lev[s]=0;
    queue<int> q; q.push(s);
    while(!q.empty())
    {
        int u=q.front(); q.pop();
        for(int i=head[u];i;i=E[i].nxt)
        {
            int v=E[i].v;
            if(lev[v]==-1&&E[i].f)
            {
                lev[v]=lev[u]+1;
                if(v==t) return 1;
                q.push(v);
            }
        }
    }
    return 0;
}

int dfs(int u,int cap)
{
    if(u==t) return cap;
    int flow=cap;
    for(int i=head[u];i;i=E[i].nxt)
    {
        int v=E[i].v;
        if(lev[v]==lev[u]+1&&E[i].f>0&&flow)
        {
            int f=dfs(v,min(E[i].f,flow));
            flow-=f;
            E[i].f-=f; E[i^1].f+=f;
        }
    }
    return cap-flow;
}

int main()
{
    n=read();p=read();q=read();t=n*2+p+q+1;
    
    for(int i=1;i<=p;++i)//超源向每个房间
    add(s,i,1),add(i,s,0);
    
    for(int i=1;i<=n;++i)//每个客人拆点保证只匹配一次
    add(i+p,i+p+n,1),add(i+p+n,i+p,0);
    
    for(int i=1;i<=q;++i)//每个菜向超汇
    add(i+2*n+p,t,1),add(t,i+2*n+p,0);
    
    for(int i=1;i<=n;++i)
    for(int j=1;j<=p;++j)//喜欢的房间->客人入点
    if(read())add(j,i+p,1),add(i+p,j,0);
    
    for(int i=1;i<=n;++i)
    for(int j=1;j<=q;++j)//客人出点->喜欢的菜
    if(read())add(i+n+p,j+2*n+p,1),add(j+2*n+p,i+n+p,0);
    
    while(bfs())
    maxf+=dfs(s,inf);
    
    printf("%d",maxf);
    return 0;
}