[ABC206E] Divide Both

[ABC206E] Divide Both

题意

​ 给定$L\le R$，计算出(x,y)的对数满足以下条件：

• $L\le x,y\le R$
• 设g是x,y的最大公约数，满足$g\ne 1,\frac{x}{g}\ne 1,\frac{y}{g}\ne 1$

分析

容斥原理

通过题意，可以将其转化为计算公式:
$$\begin{array}{r}\sum _{i=L}^R\sum _{j=L}^R[gcd(i,j)\ne 1][gcd(i,j)\ne i][gcd(i,j)\ne j]\end{array}$$
其中$L\le x\le R,L\le y\le R$,所以一共的方案数就是$(R-L+1{)}^2$
$$\begin{array}{r}\sum _{i=L}^R\sum _{j=L}^R\sum _{g=1}^R[gcd(i,j)=g]=(R-L+1{)}^2\end{array}$$
因此根据容斥原理，我们需要找出三种数

• $gcd(x,y)=1$
• $gcd(x,y)=x$
• $gcd(x,y)=y$

其中$gcd(x,y)=x$和$gcd(x,y)=y$可以合并为一种数。因此将方案总数减去每一种数的个数，再加上两种数的交集就是最终的答案数

令总方案数为$S1$，$gcd(x,y)=1$方案数为$S2$，$gcd(x,y)=x$方案数为$S3$，$gcd(x,y)=1\&\&gcd(x,y)=x$方案数为$S4$，可知答案就是$S1-S2-S3+S4$

可知总方案数$S1=(R-L+1{)}^2$

计算$S2:gcd(x,y)=1$:
$$\begin{array}{rl}S2& =\sum _{i=L}^R\sum _{j=L}^R[gcd(i,j)=1]\\ & =\sum _{i=L}^R\sum _{j=L}^R\sum _{d|x,d|y}\mu (d)\\ & =\sum _{d=1}^R\mu (d)(\lfloor \frac{R}{d}\rfloor -\lfloor \frac{L+d-1}{d}\rfloor +1{)}^2\\ & =\sum _{d=1}^R\mu (d)(\lfloor \frac{R}{d}\rfloor -\lfloor \frac{L-1}{d}\rfloor {)}^2\end{array}$$
若看不懂莫比乌斯函数的转化，可以参考帮助理解

计算$S3:gcd(x,y)=x$:

可知在$[L,R]$中为任意一个数$z$的倍数的个数为$\lfloor \frac{R}{z}\rfloor -\lfloor \frac{L+z-1}{z}\rfloor +1$

在$[1,R]$中为任意一个数$z$的倍数的个数为$\lfloor \frac{R}{z}\rfloor$
$$\begin{array}{rl}S3& =\sum _{x=L}^R\lfloor \frac{R}{x}\rfloor +\sum _{y=L}^R\lfloor \frac{R}{y}\rfloor -(R-L+1)\\ & =2\sum _{x=L}^R\lfloor \frac{R}{x}\rfloor -(R-L+1)\end{array}$$
其中$(R-L+1)$指的是重复计算的$x=y$的部分

计算$S4:gcd(x,y)=1\&\&gcd(x,y)=x$：

不难发现，当$L=1$时，x和y中必定至少有一个是1，这种情况下才需要计算$S4$，否则不需要计算，因为$x,y$根本取不到1，因此只需要计算出$x=1$和$y=1$时的方案数，减去$x=1,y=1$的方案数即可
$$\begin{array}{r}S4=2(R-L+1)-1\end{array}$$
最终答案ans:
$$\begin{array}{r}ans=S1-S2-S3+[L=1]S4\end{array}$$

AC代码

#include <bits/stdc++.h>
using namespace std;
using LL = long long;
int prime[1000010], cnt, mobius[1000010];
bool vis[1000010];
void init() {
    mobius[1] = 1;
    for (int i = 2; i <= 1000000; i++) {
        if (!vis[i]) {
            prime[++cnt] = i;
            mobius[i] = -1;
        }
        for (int j = 1; j <= cnt && i * prime[j] <= 1000000; j++) {
            vis[i * prime[j]] = true;
            if (i % prime[j] == 0) {
                break;
            }
            mobius[i * prime[j]] = -mobius[i];
        }
    }
}
int main() {
    ios::sync_with_stdio(false);
    cin.tie(nullptr);
    init();
    int l, r;
    cin >> l >> r;
    LL ans = 1LL * (r - l + 1) * (r - l + 1);
    for (int d = 1; d <= r; d++) {
        ans = ans - 1LL * mobius[d] * (r / d - (l - 1) / d) * (r / d - (l - 1) / d);
    }
    for (int d = l; d <= r; d++) {
        ans = ans - 1LL * (r / d) * 2;
    }
    ans = ans + r - l + 1;
    if (l == 1) {
        ans = ans + 2 * (r - l + 1) - 1;
    }
    cout << ans << '\n';
    return 0;
}