lyk有两序列a和b。
lyk想知道存在多少对x,y,满足以下两个条件。
1:gcd(x,y)=1。
2: abxabx = baybay 。
例如若a={1,1,1},b={1,1,1}。那么存在7对,因为除了x=2,y=2或x=3,y=3外都满足条件。
Input
第一行一个数n(1<=n<=100000)。
接下来一行n个数,表示ai(1<=ai<=n)。
接下来一行n个数,表示bi(1<=bi<=n)。
Output
一行表示答案
Sample Input
3
1 1 1
1 1 1
Sample Output
7
就是利用莫比乌斯求出gcd=1的种数,难点在于ologn的统计吧,直接利用莫比乌斯跳就好,因为值不超过1e5
#include <bits/stdc++.h>
using namespace std;
const int N = 1e5+10;
typedef long long ll;
int a[N],b[N];
int viss[N];
int n;
bool vis[N+10];
int prime[N+10],mu[N+10];
int cnt;
void Init(){
memset(prime,0,sizeof(prime));
memset(mu,0,sizeof(mu));
memset(vis,0,sizeof(vis));
mu[1] = 1;
cnt = 0;
for(int i=2; i<N; i++){
if(!vis[i]){
prime[cnt++] = i;
mu[i] = -1;
}
for(int j=0; j<cnt&&i*prime[j]<N; j++){
vis[i*prime[j]] = 1;
if(i%prime[j]) mu[i*prime[j]] = -mu[i];
else{
mu[i*prime[j]] = 0;
break;
}
}
}
}
ll cal(int x)
{
ll res=0;
for(int i=x;i<=n;i+=x)
viss[a[b[i]]]++;
for(int i=x;i<=n;i+=x)
res+=viss[b[a[i]]];
for(int i=x;i<=n;i+=x)
viss[a[b[i]]]--;
return res;
}
int main()
{
Init();
scanf("%d",&n);
for(int i=1;i<=n;i++)
scanf("%d",&a[i]);
for(int i=1;i<=n;i++)
scanf("%d",&b[i]);
ll res=0;
for(int i=1;i<=n;i++)
{
res+=1ll*mu[i]*cal(i);
}
printf("%lld\n",res );
}