CodeForces - 859E Desk Disorder 思维 + 并查集判环

传送门：Codeforces 859E

题意：有N个人和2N个座位。告诉你这N个人它们现在的座位。以及它们想去的座位。
每个人可以去它们想去的座位或者就坐在原来的座位上。
新的座位安排和旧的座位安排,都不允许一个座位被两个人占据的情况。
问你新的座位安排的方案数。

思路：将整个关系看成一张图，某个人现在的座位以及他想去的座位之间有一条有向边，那么我们可以分析的出这图的几个基本性质：

1. 图中可能有多个连通块

2. 每个点最多有一个出度

3. 由2可以得出每个连通块内部最多有一个环， 可能是一个自环，也可能是一个包含连通块所有点的环，但是不可能环上只包含部分连通块内的点。

4. 连通块还可能是一棵树

由以上分析，我们可以分别计算出每一种连通块的安排方案数，树形的连通块有结点个数种安排方法（画图），含自环的连通块只有1种安排方法（维持原样），含大环的连通块有两种安排方法（维持原样或者都去想去的位置）。

最后把所有连通块的安排方案数乘起来就是答案了。

由于每个连通块最多含一个环，因此只用并查集就可以很好的解决判环问题了。

代码：

#include<bits/stdc++.h>
#define ll long long
#define inf 0x3f3f3f3f
using namespace std;
typedef pair<int,int> P;
const int MAXN = 200010;
const int mod = 1e9 + 7;
int f[MAXN], sz[MAXN];
int getf(int k)
{
	return k == f[k] ? k : f[k] = getf(f[k]);
}
bool loop[MAXN], cycle[MAXN];
int main()
{
	int n, x, y, u, v;
	cin >> n;
	for(int i = 0; i <= 2 * n; i++)
	f[i] = i, sz[i] = 1;
	for(int i = 0; i < n; i++)
	{
		scanf("%d %d", &x, &y);
		if(x == y)
		loop[getf(x)] = 1;
		u = getf(x), v = getf(y);
		if(u != v)
		{
			f[u] = v;
			sz[v] += sz[u];
			loop[v] |= loop[u];
		}
		else
		cycle[u] = 1;
	}
	ll ans = 1;
	for(int i = 1; i <= 2 * n; i++)
	{
		if(getf(i) == i)
		{
			if(loop[i]) continue;
			if(cycle[i]) ans = ans * 2 % mod;
			else ans = ans * sz[i] % mod;
		}
	}
	cout << ans << endl;
 	return 0;
}