原文http://www.cnblogs.com/zhouzhendong/p/8665675.html
题目传送门 - 51Nod1675
题意
给定序列$a,b$,让你求满足$\gcd(x,y)=1,a_{b_x}=b_{a_y}$的$(x,y)$的个数。
题解
我们先考虑没有$gcd(x,y)=1$的情况。
仔细一看发现$a_{b_x}=b_{a_y}$是个障眼法,跟你绕来绕去。
弄个新的$A,B$序列,其中$A_x=a_{b_x},B_x=b_{a_x}$。然后就把这个条件变成了$A_x=B_y$。舒服多了。
然后我们可以把其中一个序列信息放进桶里面,然后另一个随便弄几下,就可以$O(n)$搞定了。
考虑到$gcd(x,y)=1$。于是这里要用到莫比乌斯反演套路:倍数反演。
设$f(i)$表示$i=gcd(x,y)$的满足条件的答案数。
设$F(i)$表示$i|gcd(x,y)$的满足条件的答案数。
于是这里可以放上倍数反演的式子:
$$F(n)=\sum_{n|d}f(d)\Longrightarrow f(n)=\sum_{n|d}\mu(\frac{d}{n})F(d)$$
这里只需要求$f(1)=\sum_{i=1}^{n}\mu(i)*F(i)$。
代码
#include <bits/stdc++.h>
using namespace std;
typedef long long LL;
const int N=100005;
int n,a[N],b[N],_a[N],_b[N],tax[N];
int prime[N],u[N],pcnt=0;
LL F[N];
bool f[N];
void get_prime(int n){
memset(f,true,sizeof f);
u[1]=1,f[0]=f[1]=0;
for (int i=2;i<=n;i++){
if (f[i])
prime[++pcnt]=i,u[i]=-1;
for (int j=1;j<=pcnt&&i*prime[j]<=n;j++){
f[i*prime[j]]=0;
if (i%prime[j])
u[i*prime[j]]=-u[i];
else {
u[i*prime[j]]=0;
break;
}
}
}
}
int main(){
scanf("%d",&n);
get_prime(n);
for (int i=1;i<=n;i++)
scanf("%d",&_a[i]);
for (int i=1;i<=n;i++)
scanf("%d",&_b[i]);
for (int i=1;i<=n;i++)
a[i]=_a[_b[i]],b[i]=_b[_a[i]];
memset(tax,0,sizeof tax);
LL ans=0;
for (int i=1;i<=n;i++){
F[i]=0;
for (int j=i;j<=n;j+=i)
tax[a[j]]++;
for (int j=i;j<=n;j+=i)
F[i]+=tax[b[j]];
for (int j=i;j<=n;j+=i)
tax[a[j]]--;
ans+=F[i]*u[i];
}
printf("%lld",ans);
return 0;
}