51nod 1675 序列变换

莫比乌斯反演的题目。

f(k) 为满足 abx=bay 的情况下, gcd(x,y)=k 的数量。
F(k) 为满足 abx=bay 的情况下, gcd(x,y)=k 的数量。

F(k)=k|df(d)

由莫比乌斯反演第二种形式,得到
f(k)=k|du(dk)F(d)

k=1 时,
f(1)=d=1nu(d)F(d)

u(d) 通过线性筛实现, F(d) 可以通过平均复杂度 O(n) 的暴力实现。
总时间复杂度 O(nn)

//#include<bits/stdc++.h>
#include<stdio.h>
#include<cstring>
#include<cmath>
#include<iostream>
using namespace std;

const int MAXN=100100;
int p[MAXN],mu[MAXN],sum[MAXN],f[MAXN];
int a[MAXN],b[MAXN];
int n;

long long cal(int x)
{
    long long i,ret=0;
    for(i=x;i<=n;i+=x)
        sum[a[b[i]]]++;
    for(i=x;i<=n;i+=x)
        ret+=sum[b[a[i]]];
    for(i=x;i<=n;i+=x)
        sum[a[b[i]]]--;
    return ret;
}

int main()
{
    int pcnt,i,j;
    long long ans;
    pcnt=0;
    memset(p,0,sizeof(p));
    memset(f,0,sizeof(f));
    memset(sum,0,sizeof(sum));
    mu[1]=1;
    for(i=2;i<MAXN;i++)
    {
        if(!f[i])
        {
            p[++pcnt]=i;
            mu[i]=-1;
        }
        for(j=1;p[j]*i<MAXN;j++)
        {
            f[p[j]*i]=1;
            if(i%p[j]==0)
            {
                mu[p[j]*i]=0;
                break;
            }
            mu[p[j]*i]=-mu[i];
        }
    }
    while(~scanf("%d",&n))
    {
        for(i=1;i<=n;i++)
            scanf("%d",&a[i]);
        for(i=1;i<=n;i++)
            scanf("%d",&b[i]);
        ans=0;
        for(i=1;i<=n;i++)
        {
            if(mu[i])
                ans+=mu[i]*cal(i);
        }
        printf("%lld\n",ans);
    }
}
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值