题意:
有n个王子和n个女孩,每个王子可能喜欢多个女孩。
先给出一个初始的完备匹配,问每个王子可以选择哪些女孩(即无论王子选择这些女孩中的哪一个),使得剩下的王子仍能够选择喜欢的女孩。
思路:
每个王子向自己喜欢的每个女孩连边,
对于每一对原配,女孩向王子连边。
然后tarjan求强连通分量,与每个王子在同一强连通分量里的女孩就是答案。
解释:
首先可以看出任何一对原配都可以互达。
对于每一个强连通分量,假设王子a选择了另一个女孩x,则女孩x的原配王子b一定可以找到另一个喜欢的女孩(他至少能找到王子a的原配y,
因为王子a和女孩y可以互达,王子b与王子a在同一个强连通分量,所以他和女孩y也一定在同一个强连通分量)。
代码:
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <iostream>
#include <vector>
#include <queue>
#include <stack>
using namespace std;
int n, t;
vector<int> edges[4005];
vector<int> ans[4005];
int dfn[4005], low[4005];
int vis[4005], col[4005];
stack<int> s;
int cnt = 0, num = 0;
void tarjan(int u) {
s.push(u);
vis[u] = 1;
dfn[u] = low[u] = ++cnt;
for (int i = 0; i < edges[u].size(); i++) {
int v = edges[u][i];
if (!dfn[v]) {
tarjan(v);
low[u] = min(low[u], low[v]);
} else if (vis[v])
low[u] = min(low[u], dfn[v]);
}
if (low[u] == dfn[u]) {
num++;
do {
t = s.top();
s.pop();
col[t] = num;
vis[t] = 0;
} while (t != u);
}
}
int main() {
while (~scanf("%d", &n)) {
int k, a;
for (int i = 0; i <= n; i++) {
edges[i].clear();
ans[i].clear();
}
for (int i = 1; i <= n; i++) {
scanf("%d", &k);
for (int j = 0; j < k; j++) {
scanf("%d", &a);
edges[i].push_back(a + n); // 每个王子向喜欢的女孩连边
}
}
for (int i = 1; i <= n; i++) {
scanf("%d", &a);
edges[a + n].push_back(i); // 女孩向原配的王子连边
}
memset(dfn, 0, sizeof(dfn));
memset(low, 0, sizeof(low));
memset(vis, 0, sizeof(vis));
memset(col, 0, sizeof(col));
while (!s.empty()) s.pop();
cnt = num = 0;
// tarjan求强连通分量
for (int i = 1; i <= n; i++)
if (!dfn[i]) tarjan(i);
for (int i = 1; i <= n; i++) {
for (int j = 0; j < edges[i].size(); j++) {
// 对于每个王子喜欢的女孩,如果在同一个强连通分量中,说明可以选择
if (col[i] == col[edges[i][j]])
ans[i].push_back(edges[i][j] - n);
}
sort(ans[i].begin(), ans[i].end()); // 题目要求按编号从小到大输出
}
for (int i = 1; i <= n; i++) {
int siz = ans[i].size();
printf("%d", siz);
for (int j = 0; j < siz; j++) {
printf(" %d", ans[i][j]);
}
printf("\n");
}
}
return 0;
}