AT_abc247_f [ABC247F] Cards 题解

题意

有 $N$ 张卡片，每张卡片的正面、反面都有一个数字。第 $i$ 张卡片的正面的数字为 $P_i$，反面的数字为 $Q_i$.
P = { P 1 , P 2 , … , P n } P=\{P_1, P_2, \dots, P_n\} P={P1​,P2​,…,Pn​}， Q = { Q 1 , Q 2 , … , Q n } Q = \{Q_1, Q_2, \dots, Q_n \} Q={Q1​,Q2​,…,Qn​} 都是一个 { 1 , 2 , … , N } \{1, 2, \dots, N\} {1,2,…,N} 的排列。
现在你需要从这 $N$ 张卡片中选择一些，使得牌上的数字包含 $1,2,\dots ,N$。问有多少种选择的方式。

思路

考虑使用并查集和动态规划。

我们可以将正面数字为 $i$ 和反面数字为 $i$ 的进行合并，其次我们可以用 $su{m}_i$ 表示集合 $i$ 的元素的数量。

接下来，我们要推出集合元素为 $i$ 的方案数量，我们假设用 $d{p}_i$ 表示集合元素为 $i$ 的方案数量，则答案就是为 ${\prod }_{i=1}^{n}d{p}_{su{m}_i}$。

我们可以列举几个 $i$，可以发现 $d{p}_1=1$，$d{p}_2=3$，$d{p}_3=4$，$d{p}_4=7$ 等等。我们不难看出：
$$d{p}_i=d{p}_{i-1}+d{p}_{i-2}$$

代码

#include <bits/stdc++.h>
#define int long long
using namespace std;
int father[200005];
int n;
int jl[200005];
int find(int x) { return x == father[x] ? x : father[x] = find(father[x]); }//查找父节点
int add(int x, int y) { father[find(x)] = find(y); }//并查集合并
int dp[200005];
int sum[200005];
int ans = 1;
const int mod = 998244353;
signed main() {
    scanf("%lld", &n);
    for (int i = 1, x; i <= n; i++) {
        scanf("%lld", &x);
        jl[x] = i;
        father[i] = i;//并查集初始化
    }
    for (int i = 1, x; i <= n; i++) {
        scanf("%lld", &x);
        add(jl[x], i);//并查集合并
    }
    for (int i = 1; i <= n; i++)  sum[find(i)]++;//统计数量
    dp[1] = 1;//初始化
    dp[2] = 3;//初始化
    for (int i = 3; i <= 200000; i++) dp[i] = (dp[i - 1] + dp[i - 2]) % mod;//dp
    for (int i = 1; i <= n; i++) {
        if (dp[sum[i]] == 0) //要是一个都没有，就跳过
            continue;
        ans = ((dp[sum[i]] % mod) * ans) % mod;//记得 mod
    }
    printf("%lld\n", ans);
    return 0;
}