题意:
如果b中环的长度是a中环长度的因子,答案+(a[i]*j*b[j])
跑两遍dfs找出a中长度为i环的个数记为as[i],和b中长度为j环的个数bs[j]
然后跑循环计算就可以了.
code:
#include<cstdio>
#include<cstring>
#include<iostream>
using namespace std;
const int mod=1e9+7;
typedef long long ll;
const int maxn=1e5+10;
int a[maxn];
int b[maxn];
int as[maxn];
int bs[maxn];
bool vis[maxn];
int ans=1;
void dfs(int t,int l,int *a,int *b)
{
if(vis[t])
{
b[l]++;
return ;
}
vis[t]=1;
dfs(a[t],l+1,a,b);
}
int main()
{
int n;
int m;
int kase=0;
while(scanf("%d%d",&n,&m)!=EOF)
{
memset(a,0,sizeof(a));
memset(b,0,sizeof(b));
for(int i=0;i<n;i++)
{
scanf("%d",&a[i]);
}
for(int i=0;i<m;i++)
{
scanf("%d",&b[i]);
}
memset(as,0,sizeof(as));
memset(bs,0,sizeof(bs));
memset(vis,0,sizeof(vis));
//cout<<"-------------"<<endl;
for(int i=0;i<m;i++)
{
if(!vis[i])
dfs(i,0,b,bs);
}
memset(vis,0,sizeof(vis));
for(int i=0;i<n;i++)
{
if(!vis[i])
dfs(i,0,a,as);
}
ans=1;
//cout<<"-------------"<<endl;
for(int i=1;i<=n;i++)
{
if(as[i])
{
int res=0;
for(int j=1;j*j<=i;j++)
{
if(i%j==0)
{
(res+=(ll)bs[j]%mod*j%mod)%=mod;
if(j*j!=i)
(res+=(ll)bs[i/j]%mod*(i/j)%mod)%=mod;
}
}
for(int j=1;j<=as[i];j++)
ans=(ll)ans *res%mod;
}
}
//cout<<"-------------"<<endl;
printf("Case #%d: %d\n", ++kase, ans);
}
}