题意:给你排列a和b,再给你一个c数组,里面不确定的数是0,确定的数要么是a[i]要么是b[i],求有多少种排列c。
思路:c[i]如果是1的话说明已经确定不用管了,如果是0,要么选a[i]要么选b[i],当a[i]==b[i]的时候,我们只能填a[i]一个数,只有一种情况,当a[i]!=b[i]的时候,如果选a[i]的时候就不能再选b数组中等于a[i]的数了,这两个数是有关联的,我们把有关联的数放到一个集合中,如果集合里的数个数为1的话,就是a[i]==b[i]的情况,我们就只有一种情况,如果集合里的数被确定一个的话,因为这些数都是有关联的,所以集合中只要有一个数确定了我们就能把整个集合求出来,只有一种情况,当一个集合里的数的个数既大于1,里面的数又都不确定的时候,他的方案数有两个,我们在算最后结果的时候是把每个集合看成c数组的一小部分来看的,所以所有的情况就是每一小部分的情况数相乘。
#include<iostream>
#include<cstring>
#include<algorithm>
using namespace std;
int n;
const int N=100005;
const int mod=1e9+7;
int a[N],b[N],c[N],p[N];
int find1(int x){
if(p[x]!=x)p[x]=find1(p[x]);
return p[x];
}
void sove(){
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];
}
long long con=1;
for(int i=1;i<=n;i++){
int x;
cin>>x;
if(x==0&&a[i]!=b[i]){//当c[i]这个位置没有确定并且集合里的个数大于1的时候
int da=find1(a[i]);
int db=find1(b[i]);
if(da==db){//找到两个数的父结点,如果相等说明在一个集合里面,并且没有确定过,就有两种情况
con*=2;//因为自己模拟一下就知道,如果一个集合里的数都不确定的话,那么我们最后肯定能找到两个数在这个集合里,根据这个条件来判断是否*2
con%=mod;
}else p[da]=db;//如果父结点不相等的话就是没有在一个集合,我们把他放到一个集合里
}
}
cout<<con<<endl;
}
int main(){
int t;
cin>>t;
while(t--){
sove();
}
return 0;
}