莫比乌斯反演 - YY的GCD - 洛谷 P2257

莫比乌斯反演 - YY的GCD - 洛谷 P2257

题意：

$给定N,M，求1\le x\le N，1\le y\le M且\gcd (x,y)为质数的(x,y)有多少对。$

输入格式

第一行一个整数 T 表述数据组数。

接下来 T 行，每行两个正整数，N,M。

输出格式

T 行，每行一个整数表示第 i 组数据的结果。

输入输出样例
输入

2
10 10
100 100

输出

30
2791

说明/提示

$T\le 10^4，N,M\le 10^7。$

---

分析：

$设f(d)为满足gcd(i,j)=d的数对的个数，F(n)为满足gcd(i,j)是n的倍数的数对的个数。即：$

$f(d)={\sum }_{i=1}^n{\sum }_{j=1}^m[gcd(i,j)=d]$

$F(n)={\sum }_{n|d}f(d)=\lfloor \frac{N}{n}\rfloor \cdot \lfloor \frac{M}{n}\rfloor$

$$根据反演定理:f(n)=\sum _{n|d}\mu (\lfloor \frac{d}{n}\rfloor )F(d)$$

$$Ans=\sum _{p\in prime}\sum _{i=1}^n\sum _{j=1}^m[gcd(i,j)=p]=\sum _{p\in prime}f(p)$$

$下面化简Ans：$

$我们枚举倍数d，$

$$Ans=\sum _{p\in prime}f(p)=\sum _{p\in prime}\sum _{d=1}^{min(\lfloor \frac{n}{p}\rfloor ,\lfloor \frac{m}{p}\rfloor )}\mu (d)F(dp)=\sum _{p\in prime}\sum _{d=1}^{min(\lfloor \frac{n}{p}\rfloor ,\lfloor \frac{m}{p}\rfloor )}\mu (d)\lfloor \frac{n}{dp}\rfloor \lfloor \frac{m}{dp}\rfloor$$

$令T=d\cdot p，即T是p的倍数$

$$Ans=\sum _{T=1}^{min(n,m)}\lfloor \frac{n}{T}\rfloor \lfloor \frac{m}{T}\rfloor \sum _{p|T,p\in prime}\mu (\frac{T}{p})$$

$到这里，我们就能够计算出答案了。需要预处理几个数组：$

$对于每个质数p，我们枚举它的倍数T，$

$对于每一个T，我们需要计算{\sum }_{p|T,p\in prime}\mu (\frac{T}{p})，即T的所有质因数的倍数(T/p)莫比乌斯函数的和。$

$所以我们预处理数组g。则:g[T]={\sum }_{p|T,p\in prime}\mu (\frac{T}{p})。时间复杂度O(\frac{n}{logn}\cdot logn)=O(n)$

$由于前一部分需要整除分块，我们还需要快速计算每一块的数组g的和，$

$所以，我们再对数组g求前缀和。$

$最后结合整除分块能够计算答案。$

$时间复杂度：O(n+T\sqrt{n})$

代码：

#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>

#define ll long long

using namespace std;

const int N = 1e7+10;

int primes[N], mu[N], cnt;
ll g[N], sum[N];
bool st[N];
int T, n, m;

void get_prime(int n)
{
    mu[1]=1;
    for(int i=2;i<=n;i++)
    {
        if(!st[i])
        {
            primes[cnt++]=i;
            mu[i]=-1;
        }
        for(int j=0;primes[j]*i<=n;j++)
        {
            int p=primes[j];
            st[i*p]=true;
            if(i%p==0)
            {
                mu[i*p]=0;
                break;
            }
            mu[i*p]=-mu[i];
        }
    }
    for(int i=0;i<cnt;i++)
    {
        int p=primes[i];
        for(int T=p;T<=n;T+=p)
            g[T]+=mu[T/p];
    }
    for(int i=1;i<=n;i++) sum[i]=sum[i-1]+g[i];
}

ll cal(int a, int b)
{
    int n=min(a,b);
    ll res=0;
    for(int l=1, r;l<=n;l=r+1)
    {
        r=min(a/(a/l), b/(b/l));
        r=min(n,r);
        res+=(ll)(a/l)*(b/l)*(sum[r]-sum[l-1]);
    }
    return res;
}

int main()
{
    get_prime(N-1);
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d%d",&n,&m);
        printf("%lld\n",cal(n,m));
    }
    
    return 0;
}