正題

題目連接:http://www.51nod.com/Challenge/Problem.html#problemId=1675

---

題目大意

給出兩個長度為$n$的序列$a,b$，求有多少對$x,y$滿足
$$gcd(x,y)=1且a_{b_x}=b_{a_y}$$
$1\le n\le 10^5,1\le a_i,b_i\le n$

---

解題思路

額挺明顯的一個莫反，枚舉約數$d$的時候用一個數組統計一下有多少個$a_{b_x}$就好了。

時間復雜度$O(n\log n)$

---

code

#include<cstdio> #include<cstring> #include<algorithm> #define ll long long using namespace std; const ll N=1e5+10; ll n,cnt,ans,a[N],b[N],c[N],mu[N],pri[N/10]; bool v[N]; void Prime(){mu[1]=1;for(ll i=2;i<=n;i++){if(!v[i])pri[++cnt]=i,mu[i]=-1;for(ll j=1;j<=cnt&&i*pri[j]<=n;j++){v[i*pri[j]]=1;if(i%pri[j]==0)break;mu[i*pri[j]]=-mu[i];}}return; } signed main() {scanf("%lld",&n);for(ll i=1;i<=n;i++)scanf("%lld",&a[i]);for(ll i=1;i<=n;i++)scanf("%lld",&b[i]);Prime();for(ll i=1;i<=n;i++){ll sum=0;for(ll j=i;j<=n;j+=i)c[a[b[j]]]++;for(ll j=i;j<=n;j+=i)sum+=c[b[a[j]]];for(ll j=i;j<=n;j+=i)c[a[b[j]]]--;ans+=sum*mu[i];}printf("%lld\n",ans);return 0; }

總結

以上是生活随笔為你收集整理的51nod1675-序列变换【莫比乌斯反演】的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。