原题链接:https://atcoder.jp/contests/abc247/tasks/abc247_f
题意:
给定N张牌,每张牌正反面各有一个数,所有牌的正面、反面分别构成大小为N的排列P,Q。
求有多少种摆放方式,使得N张牌朝上的数字构成一个1~N的排列。
思路:dp + 并查集
-
建图:有1~N的顶点,然后Pi跟Qi连一条边。
因为给定的是两个排列,所以每个点的度数为2,因而建出的图必然是由几个独立的环构成。
根据乘法原理,答案就等于每个环的方案数相乘。
-
求每个环的方案数:
假设环的大小为n(点的数量),dp[n]表示这样的环的方案数:dp[1] = 1,dp[2] = 3,当 n >= 3 时,dp[n] = dp[n - 1] + dp[n - 2]。
上述结论证明:因为每个点都要选出,每个点又跟两条边相连,因此选边时:若此边不选,下条边必须选;若此边已选,下条边可选可不选。那么对于n - 1个点的环,插入第n个点,就由两种情况转移过来。
-
判环及求环的大小:
并查集可以很好的完成任务 : )
(不得不说并查集真的是简洁又好用的数据结构
代码参考:
//Jakon:dp + 并查集
#include <bits/stdc++.h>
#define int long long
using namespace std;
const int mod = 998244353;
const int N = 200010;
int n, p[N], q[N], dp[N], fa[N], siz[N];
int find(int x)
{
return fa[x] == x ? x : fa[x] = find(fa[x]);
}
signed main()
{
cin >> n;
for(int i = 1; i <= n; i++) cin >> p[i];
for(int i = 1; i <= n; i++) cin >> q[i];
dp[1] = 1, dp[2] = 3;
for(int i = 3; i <= n; i++) dp[i] = (dp[i - 1] + dp[i - 2]) % mod;
for(int i = 1; i <= n; i++) fa[i] = i, siz[i] = 1;
for(int i = 1; i <= n; i++) {
int x = find(p[i]), y = find(q[i]);
if(x != y) fa[x] = y, siz[y] += siz[x];
}
int ans = 1;
for(int i = 1; i <= n; i++)
if(fa[i] == i) ans = ans * dp[siz[i]] % mod;
cout << ans << endl;
return 0;
}