题意:该酒店只有p间房间,一天只有固定的q道不同的菜。有一天来了n个客人,每个客人说出了自己喜欢哪些房间,喜欢哪道菜。但是很不幸,可能做不到让所有顾客满意(满意的条件是住进喜欢的房间,吃到喜欢的菜)。这里要怎么分配,能使最多顾客满意呢?
思路:两个二分图,一个二分图匹配房间,另一个匹配菜,但要多定义两个数组记录dfs前的状态,举个栗子:
编号为3的客人喜欢1号房间,1号菜
可是房间1号被占用了,而菜没有被占用
这时来了一个编号为4的客人,喜欢2号房(空),1号菜
这时ans不会加一,因为一号菜被占了。
所以我们可以多定义两个数组,记录dfs前的状态,如果ans不能加一,则还原dfs前的状态。
#include<bits/stdc++.h>
#define ll long long
using namespace std;
const int N = 110;
int n, p, q, hou[N][N], veg[N][N], h[N], v1[N], v2[N], v[N], l1[N], l2[N];
int F1(int x) {
for(int i = 1; i <= p; i++)
if(hou[x][i] && !v1[i]) {
v1[i] = 1;
if(h[i] == 0 || F1(h[i])) {
h[i] = x;
return 1;
}
}
return 0;
}
int F2(int x) {
for(int i = 1; i <= q; i++)
if(veg[x][i] && !v2[i]) {
v2[i] = 1;
if(v[i] == 0 || F2(v[i])) {
v[i] = x;
return 1;
}
}
return 0;
}
int main() {
cin >> n >> p >> q;
for(int i = 1; i <= n; i++)
for(int j = 1; j <= p; j++)
cin >> hou[i][j];
for(int i = 1; i <= n; i++)
for(int j = 1; j <= q; j++)
cin >> veg[i][j];
int ans = 0;
for(int i = 1; i <= n; i++) {
memset(v1, 0, sizeof v1);
memset(v2, 0, sizeof v2);
for(int j = 1; j <= p; j++)//记录状态
l1[j] = h[j];
for(int j = 1; j <= q; j++)
l2[j] = v[j];
if(F1(i) && F2(i))
ans++;
else {//还原
for(int j = 1; j <= p; j++)
h[j] = l1[j];
for(int j = 1; j <= q; j++)
v[j] = l2[j];
}
}
cout << ans;
return 0;
}