51nod 1675 序列变换 (简单莫比乌斯)

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 );
}
posted @ 2017-09-19 21:17  黑码的博客  阅读(286)  评论(0)    收藏  举报