题意:
给两个数列a(元素个数为n)和b(元素个数为m),求满足f(i) = b[f(a[i])] 的组合情况有多少种
思路:
1、数列a和b各自中的每一个数都不相同,不然函数f(x)的定义域与值域就不会满足 0 ~ n-1 与 0 ~ m-1,
2、数列a中元素a0,a1,a2,a3……an-1组成的函数 x = ay 必定存在循环,且存在 1~n 个循环组成,
分析第一个样例:
a={1,0,2} b={0,1}
1->a[1]->0->1 2 ->a[2]->2
这里有两个循环 分别为 a[0]->a[1]->a[0] 和 a[2]->a[2]
同理 数列b 也是如此
1)要满足f(i) = b[f(a[i])],枚举每一个数列 a 中的循环 x,然后找数列 b 中的循环y的个数,需要满足 y长度 是 x长度的因子;
2)由于环可以旋转,则每一个长度为len(y)的循环y 贡献的方法 为 tol(y)= num(y) * len(y),(num(y)是长度为 len(y) 的循环的个数),将该x的每一个 tol(y) 累加求和,tol(x) = tol(y1) + tol(y2) + …… + tol(yn); yi 都会 x的因子;
3) 由于不同x之间独立,ans = tol(x1) * tol(x2) + …… + tol(xn).
#include <bits/stdc++.h>
using namespace std;
#define LL long long
#define N 100005
#define M 105
#define INF 0x3f3f3f3f
#define mod 1000000007
vector<int>factor[N]; //factor[i] 存储 i 的因子
bool vis[N];
int cot_a[N],cot_b[N]; //存储 每一个 循环x, cot_b[i] = n 表示数列 b 长度为 i 的数量为 n
int a[N],b[N];
int main()
{
int n,m;
int cas = 1;
for(int i=1;i<N;i++){ //预处理 因子
for(int j=i;j<N;j+=i){
factor[j].push_back(i);
}
}
while(~scanf("%d%d",&n,&m)){
memset(vis,0,sizeof(vis));
int tol = 0;
for(int i=0;i<n;i++) scanf("%d",&a[i]);
for(int i=0;i<m;i++) scanf("%d",&b[i]);
for(int i=0;i<n;i++){
if(vis[i]) continue;
vis[i] = 1;
int cot = 1;
int w = i;
while(!vis[a[w]]){ //找循环
vis[a[w]] = 1;
w = a[w];
cot++;
}
cot_a[tol++] = cot;
}
memset(cot_b,0,sizeof(cot_b));
memset(vis,0,sizeof(vis));
for(int i=0;i<m;i++){
if(vis[i]) continue;
vis[i] = 1;
int cot = 1;
int w = i;
while(!vis[b[w]]){ //找循环
vis[b[w]] = 1;
w = b[w];
cot++;
}
cot_b[cot]++;
}
LL ans = 1;
for(int i=0;i<tol;i++){
LL w = 0;
int len = factor[cot_a[i]].size();
for(int j=0;j<len;j++){
int num = factor[cot_a[i]][j];
w += cot_b[num] * num; //循环y 贡献的方法 为 tol(y)= num(y) * len(y)
w %= mod;
}
ans = ans * w % mod;
}
printf("Case #%d: %lld\n",cas++,ans);
}
return 0;
}