poj1904 King's Quest 强连通分量

题意：

有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;
}