题意:
要求有序对 满足:
求有序对的对数。
数据范围:
分析:
令 表示
为
的倍数的有序对的对数,
表示
为
的有序对的对数。
显然有 ;
即
反演得:
当 时,有
因此,可以通过求出各个 来求得
。
代码实现:
先枚举 ,然后枚举
的倍数
,此时
,在这些数中找出哪些数对满足条件
。
代码:
#include <bits/stdc++.h>
#define pb push_back
using namespace std;
typedef long long ll;
const int N=1e5+5;
int mu[N],a[N],b[N];
ll res[N],num[N];
bool vis[N];
vector<int>prime;
void init()
{
memset(vis,0,sizeof(vis));
mu[1]=1;
for(int i=2;i<N;i++)
{
if(!vis[i])
{
prime.pb(i);
mu[i]=-1;
}
for(int j=0;j<prime.size()&&i*prime[j]<N;j++)
{
vis[i*prime[j]]=1;
if(i%prime[j]==0)
{
mu[i*prime[j]]=0;
break;
}
else
mu[i*prime[j]]=-mu[i];
}
}
}
int main()
{
int n;
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]);
for(int i=1;i<=n;i++)//枚举 i
{
for(int j=i;j<=n;j+=i)
num[a[b[j]]]++;
for(int j=i;j<=n;j+=i)
res[i]+=num[b[a[j]]];//gcd=k*i时的数对的对数,即 F(i)
for(int j=i;j<=n;j+=i)
num[a[b[j]]]=0;
}
ll ans=0;
for(int i=1;i<=n;i++)
ans+=(mu[i]*res[i]);
printf("%lld\n",ans);
return 0;
}

京公网安备 11010502036488号