题意:
要求有序对 满足:
求有序对的对数。
数据范围:
分析:
令 表示 为 的倍数的有序对的对数, 表示 为 的有序对的对数。
显然有 ;
即
反演得:
当 时,有
因此,可以通过求出各个 来求得 。
代码实现:
先枚举 ,然后枚举 的倍数 ,此时 ,在这些数中找出哪些数对满足条件 。
代码:
#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; }