疯狂 LCM（LCM SUM） - 线性筛题解

疯狂 LCM（LCM SUM） - 题解

Luogu 1891

线性筛见分割线以下。

题目回顾：

有 $T$ 组数据，对于每组数据，给定整数 $n$，计算：

$$\sum _{i=1}^n\mathrm{lcm}(i,n)$$

数据保证 $1\le T\le 3\times 10^5$，$1\le n\le 10^6$。

分析：

推式子部分

先看看能不能直接把式子抓起来算。显然这条式子不能拿来直接算（直接算意味着单次查询 $O(n)$，是不能接受的），必须推一下，使得能在合适的时间复杂度下处理。

$$\begin{array}{rl}& \sum _{i=1}^n\mathrm{lcm}(i,n)\\ =& \sum _{i=1}^n\frac{in}{\gcd (i,n)}\\ =& n\sum _{i=1}^n\frac{i}{\gcd (i,n)}\\ =& n\sum _{d\mid n}\sum _{i=1}^n\frac{i}{d}[\gcd (i,n)=d]\\ =& n\sum _{d\mid n}\sum _{i=1}^{\frac{n}{d}}i[\gcd (i,\frac{n}{d})=1]\\ =& n\sum _{d\mid n}\sum _{i=1}^{d}i[\gcd (i,d)=1]\\ =& n(\sum _{d\mid n}\frac{d\phi (d)}{2}+\frac{1}{2})\\ =& \frac{n}{2}(\sum _{d\mid n}d\phi (d)+1)\end{array}$$

推到这里就够了，有些细节需要解释一下。两个 Q&A 对应着两个难点。

Q: 倒数第三步怎么来的，为什么突然从 $n/d$ 跳跃到 $d$ 去了？
A: 我们枚举 $d$ 和枚举 $n/d$ 等价，实际上都是在找 $n$ 的约数。

Q: 我知道 ${\sum }_{i=1}^{d}i[\gcd (i,d)=1]$ 的意义是求与 $d$ 互质的数之和，之前学过它等于 $d\phi (d)/2$，为什么后面还要加一个 $1/2$？
A: 别忘了，那个结论里面 $1$ 是个例外。它的约数个数和显然是 $1$，但是按公式算得到了$1/2$。然后不管算哪一个 $n$，统计的时候都会统计 $d=1$，都会漏了 $1$ 的另外一半 $1/2$，把它补上去就行了。

假设 $f(n)={\sum }_{d\mid n}d\phi (d)$，所求就是 $n(f(n)+1)/2$，如果预处理 $f(n)$ 的值，就可以在 $O(1)$ 的时间复杂度下解决问题。

线性筛部分

观察，$n\phi (n)$ 可以看作 $id(n)$ 和 $\phi (n)$ 的积，根据积性函数的定义，我们容易知道两个积性函数的积是积性函数。

继续观察，$f(n)$ 可以看作 $n\phi (n)$ 和 $1(n)$ 两个积性函数的狄利克雷卷积，那么 $f(n)$ 就是积性函数。我们知道，只要是积性函数，就一定可以用 线性筛 在 $O(n)$ 的时间复杂度下进行预处理。

显然 $f(1)=1$。

假设 $p$ 为质数。显然 $f(p)=1+p(p-1)=p^2-p+1$。

假设线性筛外层循环枚举到 $i$。若 $p\nmid i$，显然 $f(ip)=f(i)f(p)$。

关键在于解决 $p\mid i$ 的情况。这是本题第三个难点。

---

新方案

国际金大佬曾经给出过一种方案，下面是另一种方案。截止笔者撰写此篇题解时，该方案已被较广泛使用，但尚未有人写过针对此方案的题解。现在将一步步证明出来。

先上结论：$f(ip)=f(i)+p^2(f(i)-f(i/p))$。

注意到：

$$\begin{array}{rl}& f(p^c)\\ =& \sum _{d\mid p^c}d\phi (d)\\ =& 1+\sum _{i=1}^c{p}^i\phi (p^i)\\ =& 1+\sum _{i=1}^c{p}^i\cdot p^{i-1}(p-1)\\ =& 1+\sum _{i=1}^c{p}^{2i}-p^{2i-1}\\ =& 1-p+p^2-p^3+p^4+\cdots -p^{2c-1}+p^{2c}\\ =& \sum _{i=0}^{2c}(-p{)}^i\end{array}$$

假设 $i=a\cdot p^c$，那么 $f(i)=f(a)f(p^c)$，$f(i/p)=f(a)f(p^{c-1})$，$f(ip)=f(a)f(p^{c+1})$。进而得到：

$$\begin{array}{rl}& f(ip)\\ =& f(a)f(p^{c+1})\\ =& f(a)\sum _{i=0}^{2c+2}(-p{)}^i\\ =& f(a)(\sum _{i=0}^{2c}(-p{)}^i+\sum _{i=2}^{2c+2}(-p{)}^i-\sum _{i=2}^{2c}(-p{)}^i)\\ =& f(a)(\sum _{i=0}^{2c}(-p{)}^i+p^2\sum _{i=0}^{2c}(-p{)}^i-p^2\sum _{i=0}^{2c-2}(-p{)}^i)\\ =& f(a)(f(p^c)+p^{2}f(p^c)-p^{2}f(p^{c-1}))\\ =& f(a)f(p^c)+p^2(f(a)f(p^c)-f(a)f(p^{c-1}))\\ =& f(i)+p^2(f(i)-f(i/p))\end{array}$$

容斥大法妙。

$f(ip)=f(i)+p^2(f(i)-f(i/p))$，该公式更贴合线性筛求积性函数模板，且不需要额外的改动，运行时间略优于原有方法。

代码

#include <cstdio>
#define MAXN 1000005

using namespace std;

int prime[MAXN], v[MAXN], tot;
long long f[MAXN];

void init(int n) {
    f[1] = 1;
    for (int i = 2; i <= n; ++i) {
        if (!v[i]) {
            prime[++tot] = v[i] = i;
            f[i] = 1ll * i * i - i + 1;
            // f[i] = i * i - i + 1; This will WA.
        }
        for (int j = 1; j <= tot; ++j) {
            if (prime[j] > n / i) break;
            v[i*prime[j]] = prime[j];
            if (v[i] == prime[j]) {
                f[i*prime[j]] = f[i]+(f[i]-f[i/prime[j]])*prime[j]*prime[j];
                break;
            } else {
                f[i*prime[j]] = f[i] * f[prime[j]];
            }
        }
    }
}

int main() {
    init(MAXN - 5);
    int T;
    scanf("%d", &T);
    while (T--) {
        int n;
        scanf("%d", &n);
        printf("%lld\n", (f[n] + 1) * n / 2);
    }
    return 0;
}

CSDN 的 Markdown 做的不错，以后主要发这里了。