莫比乌斯反演入门(五)--P2257(P2522升级)

题目大意:

这题是P2522 的升级版.

$T$组询问，每组询问$n,m$点对中有多少个点对满足$gcd(i,j)$为质数.

题目思路:P2522+换元+线性筛积性函数

1.前面部分直接按P2522推导即可。

但是直接算的复杂度为:$O(T\ast \pi (n)\ast \sqrt{n})$.无法接受。观察式子:

2.$原式={\sum }_{k\in p}^n{\sum }_{d=1}^{\frac{n}{d}}\mu (d)\lfloor \frac{n}{kd}\rfloor \lfloor \frac{m}{kd}\rfloor$

令$T=kd$,枚举$T$.

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

---

令$f(T)={\sum }_{k|T,k\in p}^{}\mu (\frac{T}{k})=\mu \ast 1(素数意义上的)$.

自然考虑线性筛:

$\{\begin{array}{c}f(i)=\mu (1),i\in prime\\ f(i\ast p_j)=\mu (i),i\equiv 0(mod{p}_j)\\ f(i\ast p_j)=-f(i)+\mu (i),i\ne 0(mod{p}_j)\end{array}$

推导:
1.当$i\in prime$,①显然成立。(带进式子看看即可)

2.当$i\equiv 0(mod{p}_j)$,$i\ast p_j$相较于$i$没有增加本质不同的质数。则:
$f(i\ast p_j)={\sum }_{k|i,k\in p}^{}\mu (\frac{i\ast p_j}{k})$当且仅当$k=p_j$时,$\mu$函数才有可能不为0(因为$i\ast p_j$的$p_j$的指数$\ge 2$).这时就要看$i$中有无重复质因子。即$f(i\ast p_j)=\mu (i)$.

3.当$i\ne 0(mod{p}_j)时gcd(i,p_j)=1$.则$f(i\ast p_j)={\sum }_{k|(i\ast p_j),k\in p}^{}\mu (\frac{i\ast p_j}{k})=\mu (i)+{\sum }_{k|i,k\in p}^{}\mu (\frac{i}{k}\ast p_j)$

$={\sum }_{k|(i\ast p_j),k\in p}^{}\mu (\frac{i\ast p_j}{k})=\mu (i)+\mu (p_j){\sum }_{k|i,k\in p}^{}\mu (\frac{i}{k})$

$=-f(i)+\mu (i)$

所以直接筛了。

---

回到原式，我们预处理完$f$函数前缀和之后就可以直接二维数论分块做了。

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

AC代码:

#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define pii pair<int,int>
#define pb push_back
#define mp make_pair
#define vi vector<int>
#define vll vector<ll>
#define fi first
#define se second
const int maxn = 1e7 + 5;
const int mod = 1e9 + 7;
int a[maxn];
int u[maxn] , bk[maxn] , p[maxn] , cnt;
ll f[maxn];
void init ()
{
    u[1] = 1;
    for (int i = 2 ; i < maxn ; i++){
        if (!bk[i]){
            p[++cnt] = i;
            u[i] = -1;
            f[i] = 1;
        }
        for (int j = 1 ; j <= cnt && i * p[j] < maxn ; j++){
            bk[i * p[j]] = 1;
            if (i % p[j] == 0) {
                u[i * p[j]] = 0;
                f[i * p[j]] = u[i];
                break;
            }else {
                u[i * p[j]] = -u[i];
                f[i * p[j]] = -f[i] + u[i];
            }
        }
    }
    for (int i = 1 ; i < maxn ; i++) f[i] += f[i - 1];
    return ;
}
ll solve (int n , int m){
    ll ans = 0;
    for (int l = 1 , r ; l <= n ; l = r + 1){
        r = min(n / (n / l) , m / (m / l));
        ans += 1ll * (f[r] - f[l - 1]) * (n / l) * (m / l);
    }
    return ans;
}
template <typename T>
void read(T & x){ x = 0;T f = 1;char ch = getchar();while(!isdigit(ch)){if(ch == '-') f = -1;
ch = getchar();}while (isdigit(ch)){x = x * 10 + (ch ^ 48);ch = getchar();}x *= f;}
template <typename T>
void write(T x){if(x < 0){putchar('-');x = -x;}if (x > 9)write(x / 10);putchar(x % 10 + '0');}
int main()
{
    init();
    int t; read(t);
    while (t--){
        int n , m; read(n);read(m);
        if (n > m) swap(n , m);
        printf("%lld\n" , solve (n , m));
    }
    return 0;
}