题意:有n套衣服和m个人,6种衣服的数量是一样的,每个人有两种需求,问是否能让所有人的需求都满足。
题解:明显的二分图匹配,比普通的多了一个是每种衣服的数量不是一个,而是m/6个,所以建图的时候注意把循环m/6,相当与一个人可以和 m / 6 × 2 种衣服合适,剩下就用匈牙利算法。
#include <stdio.h>
#include <string.h>
#include <string>
#include <iostream>
#include <map>
using namespace std;
const int N = 40;
int t, n, m, g[N][N], link[N], vis[N];
map<string, int> mapp;
bool dfs(int u) {
for (int i = 1; i <= m; i++) {
if (!vis[i] && g[u][i]) {
vis[i] = 1;
if (link[i] == -1 || dfs(link[i])) {
link[i] = u;
return true;
}
}
}
return false;
}
int main() {
mapp["L"] = 1;
mapp["XL"] = 2;
mapp["XXL"] = 3;
mapp["S"] = 4;
mapp["XS"] = 5;
mapp["M"] = 6;
scanf("%d", &t);
while (t--) {
scanf("%d%d", &n, &m);
string s1, s2;
int temp = n / 6;
memset(g, 0, sizeof(g));
memset(link, -1, sizeof(link));
for (int i = 1; i <= m; i++) {
cin >> s1 >> s2;
for (int j = 0; j < temp; j++) {
g[mapp[s1] + 6 * j][i] = 1;
g[mapp[s2] + 6 * j][i] = 1;
}
}
for (int i = 1; i <= n; i++) {
memset(vis, 0, sizeof(vis));
dfs(i);
}
int res = 1;
for (int i = 1; i <= m; i++) {
if (link[i] == -1) {
res = 0;
break;
}
}
if (!res)
printf("NO\n");
else
printf("YES\n");
}
return 0;
}