C. Where is the Pizza?(并查集 + 思维)

大致题意 : 有两个长度为n的排列a,b, 以及一个排列c, 排列c满足

c[i] = a[i] or b[i];

c[i]中有一些固定元素;

问排列c有几种构造方法, 答案对1e9+7取模

思路 : 观察样例, 不难发现对于排列a,b有一些关联集合, 这里取样例1进行分析

a: 1 2 3 4 5 6 7

b: 2 3 1 7 6 5 4

c: 2 0 1 0 0 0 0

可以发现对于集合 {1,2,3}, 当集合中一个元素在a或b中被选取时其他两个元素也会固定

如 第一个位置c[1] 取1 则会影响第三个位置只能取3, 进一步影响第二个位置取2, 导致这一个组合被固定, 同理第一个位置取2会存在另一种取法

• 所以每个集合会产生两种取法, 即一个集合的贡献为2
• 若位置上a,b相同, 则该集合只有一个元素, 独立不会对其他位置产生影响, 该集合贡献为0
• 若集合中某一位置在c中被确定, 则该集合也被确定, 贡献为0

记cnt为集合个数, 最终结果即为 2^cnt & mod, 因为集合间不会互相影响, 所以总种类数要乘算(集合间乘cnt次2)

对于集合的处理, 我们可以考虑使用并查集, 将每个位置的a,b进行连接, 贡献为0的特殊位置, 记录p[i] = 0

连完后求一遍并查集个数再跑一遍快速幂即可

代码及其注释:

#include <iostream>
#include <cstring>
#include <cstdio>
#include <string>
#include <iterator>
#include <cmath>
#include <algorithm>
#include <sstream>
#include <map>
#include <stack>
#include <set>
#include <queue>
#include <iomanip>
using namespace std;

stringstream ss;
typedef long long ll;
typedef pair<int, int> PII;
const int N = 5e5+10, mod = 1e9+7;
const int INF = 0x3f3f3f3f;

int a[N],b[N],c[N];
int T,p[N];

int find(int x)
{
    if(x != p[x]) p[x] = find(p[x]);
    return p[x];
}

// 快速幂模板
int qmi(int a, int k, int p)
{
    int res = 1;
    while(k)
    {
        if(k & 1) res = (ll) res*a % p;
        k >>= 1;
        a = (ll)a*a % p;
    }
    return res;
}

int main()
{
    ios::sync_with_stdio(false),cin.tie(0),cout.tie(0);
    cin>>T;
    while(T -- )
    {
        int n;
        cin>>n;
        for(int i = 1; i<=n; i++) cin>>a[i], p[i] = i;
        for(int i = 1; i<=n; i++)
        {
            cin>>b[i];
            // 对a,b用并查集合并, 相同元素贡献为0
            if(a[i] == b[i]) p[a[i]] = 0;
            int pa = find(a[i]), pb = find(b[i]);
            p[pa] = pb;
        }
        for(int i = 1; i<=n; i++)
        {
            cin>>c[i];
            // c中出现元素的集合贡献为0
            p[find(c[i])] = 0;
        }
        int cnt = 0;
        for(int i = 1; i<=n; i++) if(p[i] == i) cnt ++;
        cout<<qmi(2,cnt,mod)<<endl;
    }
}