洛谷P1402 酒店之王

题目描述

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

有一天来了n个客人，每个客人说出了自己喜欢哪些房间，喜欢哪道菜。但是很不幸，可能做不到让所有顾客满意（满意的条件是住进喜欢的房间，吃到喜欢的菜）。

这里要怎么分配，能使最多顾客满意呢？

输入输出格式

输入格式：

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

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

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

输出格式：

最大的顾客满意数。

输入输出样例

输入样例#1： 

2 2 2
1 0
1 0
1 1
1 1

输出样例#1： 

1

3个点集的二分图最大匹配，同样是匈牙利。。。

只是要注意：

当一个人匹配了房间但不匹配菜（或反之），一定要将这个人的匹配撤回，好让其他人完成2个匹配。。。

为了这个WA了10+次。。。

附代码：

#include<iostream>
#include<algorithm>
#include<cstdio>
#include<cstring>
#define MAXN 110
using namespace std;
int n,m,k,ans=0,f1[MAXN],f2[MAXN],last[MAXN];
bool vis[MAXN],a[MAXN][MAXN],b[MAXN][MAXN];
inline int read(){
       int date=0,w=1;char c=0;
       while(c<'0'||c>'9'){if(c=='-')w=-1;c=getchar();}
       while(c>='0'&&c<='9'){date=date*10+c-'0';c=getchar();}
       return date*w;
}
bool find1(int x){
     for(int i=1;i<=m;i++)
     if(!vis[i]&&a[x][i]){
                 vis[i]=true;
                 if(f1[i]==-1||find1(f1[i])){
                                             f1[i]=x;
                                             return true;
                                             }
                 }
     return false;
}
bool find2(int x){
     for(int i=1;i<=k;i++)
     if(!vis[i]&&b[x][i]){
                 vis[i]=true;
                 if(f2[i]==-1||find2(f2[i])){
                                             f2[i]=x;
                                             return true;
                                             }
                 }
     return false;
}
int main(){
    n=read();m=read();k=read();
    memset(f1,-1,sizeof(f1));
    memset(f2,-1,sizeof(f2));
    memset(a,false,sizeof(a));
    memset(b,false,sizeof(b));
    memset(vis,false,sizeof(vis));
    for(int i=1;i<=n;i++)
    for(int j=1;j<=m;j++){
            int x=read();
            if(x)a[i][j]=true;
            }
    for(int i=1;i<=n;i++)
    for(int j=1;j<=k;j++){
            int x=read();
            if(x)b[i][j]=true;
            }
    for(int i=1;i<=n;i++){
            for(int j=1;j<=k;j++)last[j]=f2[j];
            memset(vis,false,sizeof(vis));
            if(!find2(i))continue;
            memset(vis,false,sizeof(vis));
            if(find1(i))ans++;
            else for(int j=1;j<=k;j++)f2[j]=last[j];
            }
    printf("%d\n",ans);
    return 0;
}