【题解】BZOJ 2005 [Noi2010]能量采集

传送门

$Description$

根据题目描述以及最基础的找规律（真的很基础），答案就是

$$\sum_{i = 1}^{n} \sum_{j = 1}^{m}(2\gcd(i, j) - 1) = 2\sum_{i = 1}^{n} \sum_{j = 1}^{m}\gcd(i, j) -nm$$

$Solution$

其实问题的核心就在于求解

$$\sum_{i=1}^{n}\sum_{j=1}^{m}\gcd(i, j)$$
我们可以令 $f(x)$ 表示 $\gcd(i, j)$ 为 $x$ 的个数。

所以我们有

$$t = \min\{n, m\} \\ \sum_{i=1}^{n}\sum_{j=1}^{m}\gcd(i, j) = \sum_{k = 1}^t f(k)$$
当然这个可以用 Mobius 反演在 $O(n \sqrt{n})$ 的时间内得到解决，但这里讲一种 $O(n\lg n)$ 的做法。

考虑如何计算 $f(x)$

找到 $1 \cdots n$ 及 $1 \cdots m$ 中所有能整除 $x$ 的数组成的数对， 答案就是

$$\left [ \frac{n}{x} \right] \cdot \left[ \frac{m}{x} \right]$$
但是可以想到这些数对中可能有 $\gcd$ 为 $x$ 的倍数的。

举个例子，令 $n = 7, m = 8, x = 2$ ，满足 $x | i$ 且 $x | j$ 的数对 $(i, j)$ 有：

$$(2, 2), (2, 4), (2, 6), (2, 8)\\ (4, 2), (4, 4),(4, 6), (4, 8)\\ (6, 2),(6, 4), (6, 6), (6, 8)$$
共 $12$ 对，但是可以枚举得到 $f(2) = 9$ ，满足的数对有：
$$(2, 2), (2, 4), (2, 6), (2, 8), (4, 2), (4, 6), (6, 2),(6, 4), (6, 8)$$
这是为什么呢？因为 $\gcd(4,4)=4, \gcd(4, 8)=4, \gcd(6, 6) = 6$ 。这些 $\gcd$ 值都是 $x$ 的倍数。

而且观察到 $f(4)=2, f(6)=1$ 。

这样我们就可以这样计算 $f(x)$

$$f(x) =\left [ \frac{n}{x} \right] \cdot \left[ \frac{m}{x} \right] - \sum_{i=2}^{ \left[ \frac{t}{x} \right] } f(i\cdot x)$$
后面的部分就是将那些 $\gcd$ 为 $x$ 的倍数的数都删掉。

写个伪代码吧

$$\renewcommand{\tab}[1]{\hskip{#1 em}\hskip{#1 em}} \renewcommand{\func}{\text} \renewcommand{\for}{\textbf{for} ~~} \renewcommand{\to}{~~ \textbf{to} ~~} \renewcommand{\downto}{~~ \textbf{downto} ~~} \renewcommand{\return}{\textbf{return} ~~} \renewcommand{\div}{~~ \textbf{div} ~~} \renewcommand{\type}[1]{\textbf {#1} ~} \begin{aligned} & \tab{0} \func{Solve}(\type{int} n, \type{int} m) \\ & \tab{1} ans \leftarrow 0 \\ & \tab{1} t \leftarrow \min \{n, m\} \\ & \tab{1} \for s \leftarrow t \downto 1 \\ & \tab{2} f[s] \leftarrow (n \div s) \times (m \div s) \\ & \tab{2} \for i \leftarrow 2 \to t \div s \\ & \tab{3} f[s] \leftarrow f[s] - f[i \times s] \\ & \tab{2} ans \leftarrow ans + f[s] \times s \\ & \tab{1} \return ans \end{aligned}$$
Tex 代码如下：

\renewcommand{\tab}[1]{\hskip{#1 em}\hskip{#1 em}} 
\renewcommand{\func}{\text}
\renewcommand{\for}{\textbf{for} ~~}
\renewcommand{\to}{~~ \textbf{to} ~~}
\renewcommand{\downto}{~~ \textbf{downto} ~~}
\renewcommand{\return}{\textbf{return} ~~}
\renewcommand{\div}{~~ \textbf{div} ~~}
\renewcommand{\type}[1]{\textbf {#1} ~}
\begin{aligned}
& \tab{0} \func{Solve}(\type{int} n, \type{int} m) \\
& \tab{1} ans \leftarrow 0 \\
& \tab{1} t \leftarrow \min \{n, m\} \\
& \tab{1} \for s \leftarrow t \downto 1 \\
& \tab{2} f[s] \leftarrow (n \div s) \times (m \div s) \\
& \tab{2} \for i \leftarrow 2 \to  t \div s \\
& \tab{3} f[s] \leftarrow f[s] - f[i \times s] \\
& \tab{2} ans \leftarrow ans + f[s] \times s \\
& \tab{1} \return ans
\end{aligned}

$Code$

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#define min(x, y) ((x) < (y) ? (x) : (y))
long long f[100005];
int main() {
    long long n, m, ans = 0;
    scanf("%lld%lld", &n, &m);
    long long t = min(n, m);
    for (long long s = t; s >= 1; s--) {
        f[s] = (n / s) * (m / s);
        for (long long i = 2; i * s <= t; i++) 
            f[s] -= f[i * s];
    }
    for (long long i = 1; i <= t; i++)
        ans += f[i] * i;
    printf("%lld\n", 2 * ans - n * m);
    return 0;
}