[51Nod 1584] 加权约数和（约数和函数性质 + 莫比乌斯反演） | 错题本

题目

[51Nod 1584] 加权约数和

分析

先把 max ⁡ { i , j } \max\{i, j\} max{i,j} 去掉：
$$原式=2\sum _{i=1}^N\sum _{j=1}^{i}i\sigma (ij)-\sum _{i=1}^{N}i\sigma (i^2)$$

---

推式子前先介绍约数和函数有一个重要性质（和约数个数一样可以转化为 $\gcd$ 问题）：$$\begin{array}{rl}\sigma (xy)=& \sum _{i|x}\sum _{j|y}ij\left[\gcd \left(i,\frac{y}{j}\right)=1\right]\\ =& \sum _{i|x}\sum _{j|y}i\frac{y}{j}[\gcd (i,j)=1]\end{array}$$ 证明时考虑我们从 $x$ 中枚举一个因数 $i$，从 $y$ 中枚举一个因数 $j$，为了确保不算重，我们规定同一个质因子先从 $y$ 中取到 $j$ 里面，取完了再从 $x$ 中取到 $i$ 里面，这样的话所有 $ij$ 都是 $xy$ 的因数且不会算重。如何保证每个质因子都是先从 $y$ 中取的呢？只需要看 $\frac{y}{j}$ 中的某个质因数在不在 $i$ 中，如果在，就意味着 $y$ 中的这个质因数还没有取完，我们就从 $x$ 中取了，这是不合法的，于是第一个等号得证，第二个等号就显然了。

---

先算第一个部分，代入上面的公式：$$\begin{array}{rl}& \sum _{i=1}^N\sum _{j=1}^{i}i\sigma (ij)\\ =& \sum _{i=1}^N\sum _{j=1}^{i}i\sum _{x|i}\sum _{y|j}x\frac{j}{y}[\gcd (x,y)=1]\end{array}$$ 反演一下上式变为 $$\begin{array}{rl}& \sum _{i=1}^N\sum _{j=1}^{i}i\sum _{x|i}\sum _{y|j}x\frac{j}{y}\sum _{d|\gcd (x,y)}\mu (d)\\ =& \sum _{d=1}^N\mu (d)\sum _{i=1}^{\lfloor \frac{N}{d}\rfloor }\sum _{j=1}^{i}di\sum _{d|dx|di}\sum _{d|dy|dj}dx\frac{dj}{dy}\\ =& \sum _{d=1}^N\mu (d)\sum _{i=1}^{\lfloor \frac{N}{d}\rfloor }\sum _{j=1}^{i}di\sum _{x|i}\sum _{y|j}dx\frac{dj}{dy}\\ =& \sum _{d=1}^N\mu (d)\sum _{i=1}^{\lfloor \frac{N}{d}\rfloor }\sum _{j=1}^{i}di\sum _{x|i}dx\sum _{y|j}\frac{j}{y}\\ =& \sum _{d=1}^N\mu (d)\sum _{i=1}^{\lfloor \frac{N}{d}\rfloor }{d}^{2}i\sum _{x|i}x\sum _{j=1}^i\sum _{y|j}y\\ =& \sum _{d=1}^N\mu (d)d^2\sum _{i=1}^{\lfloor \frac{N}{d}\rfloor }i\sigma (i)\sum _{j=1}^i\sigma (j)\end{array}$$ 令 $S(x)=\sum _{i=1}^x\sigma (i)$，别忘了还要最后乘个 $2$： $$2\sum _{d=1}^N\mu (d)d^2\sum _{i=1}^{\lfloor \frac{N}{d}\rfloor }i\sigma (i)S(i)$$ 再考虑第二部分，继续代入公式，差不多的方法：$$\begin{array}{rl}& \sum _{i=1}^{N}i\sigma (i^2)\\ =& \sum _{i=1}^{N}i\sum _{x|i}\sum _{y|j}x\frac{i}{y}[\gcd (x,y)=1]\\ =& \sum _{d=1}^N\mu (d)\sum _{i=1}^{\lfloor \frac{N}{d}\rfloor }di\sum _{x|i}dx\sum _{y|i}\frac{i}{y}\\ =& \sum _{d=1}^N\mu (d)d^2\sum _{i=1}^{\lfloor \frac{N}{d}\rfloor }i{\sigma }^2(i)\end{array}$$ 两部分合起来：$$\begin{array}{rl}原式=& 2\sum _{d=1}^N\mu (d)d^2\sum _{i=1}^{\lfloor \frac{N}{d}\rfloor }i\sigma (i)S(i)-\sum _{d=1}^N\mu (d)d^2\sum _{i=1}^{\lfloor \frac{N}{d}\rfloor }i{\sigma }^2(i)\\ =& \sum _{d=1}^N\mu (d)d^2\sum _{i=1}^{\lfloor \frac{N}{d}\rfloor }i\sigma (i)(2S(i)-\sigma (i))\end{array}$$ 如果询问少的话可以 $O(N+T\sqrt{N})$ 做。

---

其实所有形如 $$A(N)=\sum _{i=1}^{N}f(i)\sum _{j=1}^{\lfloor \frac{N}{i}\rfloor }g(j)=\sum _{i=1}^N\sum _{j=1}^{\lfloor \frac{N}{i}\rfloor }f(i)g(j)$$ 的式子在已知 $f,g$ 时都可以 $O(N\ln N)$ 预处理所有的 $A$ 值。因为它能递推：$$A(N)=A(N-1)+(f\ast g)(N)$$ 考虑 $A(N)$ 比 $A(N-1)$ 多出来的东西，无非是当 $i|N$ 时，$\lfloor \frac{N}{i}\rfloor$ 多了一个 $g\left(\frac{N}{i}\right)$，因此 $$A(N)=A(N-1)+\sum _{i|N}f(i)g\left(\frac{N}{i}\right)=A(N-1)+(f\ast g)(N)$$ 然后我们用调和级数复杂度算出所有 $f\ast g$，$A$ 就是他们的前缀和！

---

换到这道题上，$f(x)=\mu (x)x^2,g(x)=x\sigma (x)(2S(x)-\sigma (x))$，先线筛 $\mu$ 和 $\sigma$，然后预处理即可。

代码

#include <bits/stdc++.h>

int Read() {
    int x = 0; char c = getchar();
    while (c < '0' || c > '9')
        c = getchar();
    while (c >= '0' && c <= '9')
        x = x * 10 + (c ^ 48), c = getchar();
    return x;
}

typedef long long LL;

const int MAXN = 1000000;
const int MOD = 1000000007;

inline int Add(int x, const int &y) {
    return ((x += y) >= MOD) ? (x - MOD) : x;
}

inline int Mul(const int &x, const int &y) {
    return (LL)x * y % MOD;
}

inline int Sub(int x, const int &y) {
    return ((x -= y) < 0) ? (x + MOD) : x;
}

int Mu[MAXN + 5];
LL Sig[MAXN + 5], Low[MAXN + 5];
bool Vis[MAXN + 5];
std::vector<int> Pri;

int Ans[MAXN + 5], S[MAXN + 5];

void Init(int n) {
    Mu[1] = Sig[1] = Low[1] = S[1] = 1;
    for (int i = 2; i <= n; i++) {
        if (!Vis[i])
            Mu[i] = MOD - 1, Sig[i] = Low[i] = 1 + i, Pri.push_back(i);
        for (int j = 0, k; j < (int)Pri.size() && i * Pri[j] <= n; j++) {
            Vis[k = i * Pri[j]] = true;
            if (i % Pri[j] == 0) {
                Mu[k] = 0;
                Low[k] = 1 + Low[i] * Pri[j];
                Sig[k] = Sig[i] / Low[i] * Low[k];
                break;
            }
            Mu[k] = MOD - Mu[i];
            Low[k] = 1 + Pri[j];
            Sig[k] = Sig[i] * Low[k];
        }
        S[i] = Add(S[i - 1], Sig[i] % MOD);
    }
    for (int i = 1; i <= n; i++)
        for (int j = i; j <= n; j += i)
            Ans[j] = Add(Ans[j], Mul(Mu[i], Mul(i, Mul(i, Mul(j / i, Mul(Sig[j / i], Sub(Mul(2, S[j / i]), Sig[j / i])))))));
    for (int i = 1; i <= n; i++)
        Ans[i] = Add(Ans[i], Ans[i - 1]);
}

int main() {
    Init(MAXN);
    int T = Read(), Cas = 0;
    while (T--)
        printf("Case #%d: %d\n", ++Cas, Ans[Read()]);
    return 0;
}