【算法讲9：莫比乌斯函数及其反演（实践部分：详细入门）】

好家伙，好多期的理论，终于可以实战一些题目了！

前置

• 【算法讲8：莫比乌斯函数及其反演（理论部分） | 欧拉筛】

补充

• 莫比乌斯反演，就是有算数函数 $f(n)$ 和 $F(n)$，满足 $F(n)=\sum _{d|n}f(d)$ 或者 $F(n)=\sum _{n|d}f(d)$
  然后，我们容易得到 $F(n)$，却不容易得到 $f(n)$，我们就可以用莫比乌斯反演得出 $f(n)$
• 当然，相同的题目做多了总有固定的套路，一道题目也有多种解法。

例题一：[HAOI2011]Problem b | 洛谷 P2522

• 【题目描述】洛谷 P2522
  给定正整数 $a、b、c、d、k$
  求 ： $\underset{i=a}{\sum ^b}\underset{j=c}{\sum ^d}[\gcd (i,j)=k]$
• 【数据范围】
  样例组数 $T\le 5\times 10^4$
  $1\le a\le b\le 5\times 10^4$
  $1\le c\le d\le 5\times 10^4$
• 【思路】第一题会讲的细一点
  暴力做的话时间复杂度 $O(T\times b\times d\times \log )$，我们需要数论化简。
  考虑到原式求和上下标较复杂，我们考虑化简：
  （1）$\underset{i=a}{\sum ^b}\underset{j=c}{\sum ^d}[\gcd (i,j)=k]$ 简记为 $Ans(a,b,c,d)$，由容斥定理得到：
  $Ans(a,b,c,d)=Ans(1,b,1,d)-Ans(1,b,1,c-1)-Ans(1,a-1,1,d)+Ans(1,a-1,1,c-1)$
  （2）我们专注于化简 $Ans(1,m,1,n)：$

做法一：莫比乌斯反演

• 我们设 $f(x)=\underset{i=1}{\sum ^m}\underset{j=1}{\sum ^n}[\gcd (i,j)=x]$，即所有对中，$\gcd (i,j)=x$ 的数量。
• 我们设 $F(x)=\sum _{x|d}f(d)$，即所有对中，$\gcd (i,j)$ 为 $x$ 及其倍数的数量。
  容易想到，$F(x)$ 我们非常容易算出来。
  为什么呢？因为只要求出 $i\in [1,m]$，有多少数字有因数 $x$ ，$j\in [1,n]$，有多少数字有因数 $x$
  易得 $F(x)=\lfloor \frac{m}{x}\rfloor \lfloor \frac{n}{x}\rfloor$
  我们答案是什么？就是 $f(k)$。
• 接下来，根据莫比乌斯反演的第二种方法(见前置连接)，已知 $F(x)$ 去求 $f(k)$：
  $$\begin{array}{rl}f(k)& =\sum _{k|d}\mu (\frac{d}{k})F(d)\\ & =\sum _{k|d}\mu (\frac{d}{k})\lfloor \frac{m}{d}\rfloor \lfloor \frac{n}{d}\rfloor \\ & =\underset{t=1}{\sum ^{\min (m,n)}}\mu (t)\lfloor \frac{m}{tk}\rfloor \lfloor \frac{n}{tk}\rfloor \\ & =\underset{t=1}{\sum ^{\min (\lfloor \frac{m}{k}\rfloor ,\lfloor \frac{n}{k}\rfloor )}}\mu (t)\lfloor \frac{m}{tk}\rfloor \lfloor \frac{n}{tk}\rfloor \square \end{array}$$
  第一行： 莫比乌斯反演第二个公式。
  第二行： 带入 $F(d)$
  第三行： 我们枚举 $t=\frac{d}{k}$，得到 $d=tk$，带入即可。
  第四行： 上标等值变换。
  最后，整除分块即可。我们需要用到 莫比乌斯前缀和 ，数据范围较小，用欧拉筛筛一遍即可。

做法二：套路做法

• 根据一些套路，我们有这样的做法：
  $$\begin{array}{rl}\underset{i=1}{\sum ^m}\underset{j=1}{\sum ^n}[\gcd (i,j)=k]& =\underset{i=1}{\sum ^{\lfloor \frac{m}{k}\rfloor }}\underset{j=1}{\sum ^{\lfloor \frac{n}{k}\rfloor }}[\gcd (i,j)=1]\\ & =\underset{i=1}{\sum ^{\lfloor \frac{m}{k}\rfloor }}\underset{j=1}{\sum ^{\lfloor \frac{n}{k}\rfloor }}\sum _{d|\gcd (i,j)}\mu (d)\\ & =\underset{d=1}{\sum ^{\min (\lfloor \frac{m}{k}\rfloor ,\lfloor \frac{n}{k}\rfloor )}}\mu (d)\underset{i=1}{\sum ^{\lfloor \frac{m}{k}\rfloor }}[d|i]\underset{j=1}{\sum ^{\lfloor \frac{n}{k}\rfloor }}[d|j]\\ & =\underset{d=1}{\sum ^{\min (\lfloor \frac{m}{k}\rfloor ,\lfloor \frac{n}{k}\rfloor )}}\mu (d)\lfloor \frac{m}{kd}\rfloor \lfloor \frac{n}{kd}\rfloor \square \end{array}$$
  等等…怎么就证出来了？我们细看每一行式子的推导过程。
  第一行： 这是一个等值变换，相当于问：原来 $i=[1,m]$，我们把所有有 $k$ 这个因子的数字拿出来，然后除以 $k$ ，相当于：若 $k|i$，那么 $i\stackrel{变为}{\to }\frac{i}{k}$；对于 $j$ 也是同理。然后，原来 $\gcd (i,j)=k$ 的数字都会变成 $\gcd (\frac{i}{k},\frac{j}{k})=\frac{\gcd (i,j)}{k}=1$
  第二行： 根据上一节的套路，我们已知了 $[\gcd (i,j)=1]=\epsilon (\gcd (i,j))=\sum _{d|\gcd (i,j)}\mu (d)$
  第三行： 更改求和顺序，枚举每一对 $\gcd (i,j)$ 的因子，相当于枚举每个因子，然后枚举每一对都是因子的倍数的 $(i,j)$ 对。
  第四行： $1\sim \lfloor \frac{m}{k}\rfloor$有多少个数字满足有某个因子 $d$ ？很简单，数字个数除以 $d$ 然后向下取整即可。
  然后根据该式子：$\lfloor \frac{\lfloor \frac{m}{k}\rfloor }{d}\rfloor =\lfloor \frac{m}{kd}\rfloor$

核心代码

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

/*
 _            __   __          _          _
| |           \ \ / /         | |        (_)
| |__  _   _   \ V /__ _ _ __ | |     ___ _
| '_ \| | | |   \ // _` | '_ \| |    / _ \ |
| |_) | |_| |   | | (_| | | | | |___|  __/ |
|_.__/ \__, |   \_/\__,_|_| |_\_____/\___|_|
        __/ |
       |___/
*/
const int MAX = 5e4+50;

bool vis[MAX];
ll prime[MAX];
ll mu[MAX];
ll premu[MAX];
int cnt;
void shai(int n){					/// 筛莫比乌斯函数
    mu[1] = 1;
    for(int i = 2;i <= n;++i){
        if(!vis[i]){
            prime[++cnt] = i;
            mu[i] = -1;
        }
        for(int j = 1;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;
            }
        }
    }
    for(int i = 1;i <= n;++i){		/// 求出莫比乌斯前缀和
        premu[i] = premu[i-1] + mu[i];
    }
}
ll solve(ll m,ll n,ll k){			/// 等价于前文 Ans(1,m,1,n)
    m = m / k;
    n = n / k;
    ll res = 0;
    ll L = 1;
    ll R = 0;
    while(L <= min(n,m)){
        R = min(n/(n/L),m/(m/L));
        res += (premu[R] - premu[L-1]) * (m / L) * (n / L);
        L = R + 1;
    }
    return res;
}
int main()
{
    shai(50000);
    int T;scanf("%d",&T);
    while(T--){
        ll a,b,c,d,k;
        scanf("%lld%lld%lld%lld%lld",&a,&b,&c,&d,&k);
        ll res = 0;					/// 容斥
        res += solve(b,d,k);
        res -= solve(a-1,d,k);
        res -= solve(b,c-1,k);
        res += solve(a-1,c-1,k);
        printf("%lld\n",res);
    }
    return 0;
}

---

例题二：YY的GCD | 洛谷 P2257

• 【题目描述】洛谷 P2257
  求 $\underset{i=1}{\sum ^n}\underset{j=1}{\sum ^m}[\gcd (i,j)=Prime]$
• 【数据范围】
  样例组数 $T\le 10^4$
  $1\le n,m\le 10^7$
• 【思路】
  我们使用套路求法，做题可能会更快
  $$\begin{array}{rl}\sum _{p\in Prime}\underset{i=1}{\sum ^n}\underset{j=1}{\sum ^m}[\gcd (i,j)=p]& =\sum _{p\in Prime}\underset{i=1}{\sum ^{\lfloor \frac{n}{p}\rfloor }}\underset{j=1}{\sum ^{\lfloor \frac{m}{p}\rfloor }}[\gcd (i,j)=1]\\ & =\sum _{p\in Prime}\underset{i=1}{\sum ^{\lfloor \frac{n}{p}\rfloor }}\underset{j=1}{\sum ^{\lfloor \frac{m}{p}\rfloor }}\sum _{d|\gcd (i,j)}\mu (d)\\ & =\sum _{p\in Prime}\underset{d=1}{\sum ^{\min (\lfloor \frac{n}{p}\rfloor ,\lfloor \frac{m}{p}\rfloor )}}\mu (d)\lfloor \frac{n}{pd}\rfloor \lfloor \frac{m}{pd}\rfloor \\ & =\underset{T=1}{\sum ^{\min (n,m)}}\sum _{p|T,p\in Prime}\mu (\frac{T}{p})\lfloor \frac{n}{T}\rfloor \lfloor \frac{m}{T}\rfloor \\ & =\underset{T=1}{\sum ^{\min (n,m)}}\lfloor \frac{n}{T}\rfloor \lfloor \frac{m}{T}\rfloor \sum _{p|T,p\in Prime}\mu (\frac{T}{p})\square \end{array}$$
  第一行： 我们依次枚举每一个质数 $p$
  第二行到第三行与前面套路相同。
  第四行： 考虑到我们枚举 $p$ 又枚举 $p$ 的所有倍数 $pd$，我们干脆直接设 $T=pd$，然后去枚举每一个 $T$。当然其中要求要有一个质因子 $p$ ，此时 $d=\frac{T}{p}$，把所有字母替换。
  第五行： 由于分块部分 $\lfloor \frac{n}{T}\rfloor \lfloor \frac{m}{T}\rfloor$ 与 $p$ 无关了，我们把该项提到最前面。
  注意到最后一个部分 $\sum _{p|T,p\in Prime}\mu (\frac{T}{p})$ ，我们可以暴力枚举每一个 $p$，然后对于所有 $p$ 的倍数 $pk$，我们都加上一个 $\mu (\frac{pk}{p})$，然后求一个前缀和即可参与分块运算。
  还有一点，这题大部分变量建议开 $int$ ，前缀和比较大应该开 $ll$，否则会 $TLE$

核心代码

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

/*
 _            __   __          _          _
| |           \ \ / /         | |        (_)
| |__  _   _   \ V /__ _ _ __ | |     ___ _
| '_ \| | | |   \ // _` | '_ \| |    / _ \ |
| |_) | |_| |   | | (_| | | | | |___|  __/ |
|_.__/ \__, |   \_/\__,_|_| |_\_____/\___|_|
        __/ |
       |___/
*/
const int MAX = 1e7+50;

bool vis[MAX];
int prime[MAX];
int mu[MAX];
int ff[MAX];
ll pre[MAX];
int cnt;
void shai(int n){
    mu[1] = 1;
    for(int i = 2;i <= n;++i){
        if(!vis[i]){
            prime[++cnt] = i;
            mu[i] = -1;
        }
        for(int j = 1;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;
            }
        }
    }
    for(int i = 1;i <= cnt;++i){
        int t = prime[i];
        while(t <= n){
            ff[t] += mu[t/prime[i]];
            t += prime[i];
        }
    }
    for(int i = 1;i <= n;++i)
        pre[i] = pre[i-1] + (ll)ff[i];
}
ll solve(int m,int n){
    ll res = 0;
    int L = 1;
    int R = 0;
    while(L <= min(n,m)){
        R = min(n/(n/L),m/(m/L));
        res += (pre[R] - pre[L-1]) * (m / L) * (n / L);
        L = R + 1;
    }
    return res;
}
int main()
{
    shai(10000000);
    int T;scanf("%d",&T);
    while(T--){
        int n,m;scanf("%d%d",&n,&m);
        printf("%lld\n",solve(n,m));
    }
    return 0;
}

• 用时：11.76s
  

优化

• 难道 $\sum _{p|T,p\in Prime}\mu (\frac{T}{p})$ 只能 $O(n\log n)$ 暴力算吗！事实表明不是这样。
  设 $f(i)=\sum _{p|i,p\in Prime}\mu (\frac{i}{p})$
  我们看看能否欧拉筛它！
• 假设 $p$ 是 $n$ 的最小质因数，假设 $k$ 是质数。
  （1）$f(p)=\mu (1)=1$
  （2）$n=p\times i$，当 $p\perp i$时 $f(i)=\sum _{k|i}\mu (\frac{i}{k})$
  考虑到 $f(p\times i)=\sum _{k|p\times i}\mu (\frac{p\times i}{k})=\sum _{k|i}\mu (\frac{i\times p}{k})+\mu (\frac{i\times p}{p})=-f(i)+\mu (i)$
  （3）$n=p\times i$，当 $p/\perp i$时
  考虑到当 $k\ge 2$ 时 $\mu (p^k)=0$
  当 $i$ 的最小质因子 $p$ 只有一个时，只有 $\mu (\frac{p\times i}{p})=\mu (i)$ ，其他 $\mu (\frac{p\times i}{k})=0$，故此时 $f(n)=\mu (i)$
  当 $i$ 的最小质因子 $p$ 不止一个时，所有 $\mu (\frac{p\times i}{k})=0$，不过此时 $\mu (i)=0$，也满足 $f(n)=0=\mu (i)$

优化修改处代码

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

const int MAX = 1e7+50;

bool vis[MAX];
int prime[MAX];
int mu[MAX];
int ff[MAX];
ll pre[MAX];
int cnt;
void shai(int n){
    mu[1] = 1;
    ff[1] = 0;
    for(int i = 2;i <= n;++i){
        if(!vis[i]){
            prime[++cnt] = i;
            mu[i] = -1;
            ff[i] = 1;
        }
        for(int j = 1;j <= cnt && i * prime[j] <= n;++j){
            vis[i*prime[j]] = 1;
            if(i % prime[j]){
                mu[i * prime[j]] = -mu[i];
                ff[i * prime[j]] = -ff[i] + mu[i];
            }else {
                mu[i * prime[j]] = 0;
                ff[i * prime[j]] = mu[i];
                break;
            }
        }
    }
    for(int i = 1;i <= n;++i)
        pre[i] = pre[i-1] + (ll)ff[i];
}

• 用时：8.47s