SPOJ 5971 LCM SUM

转载自 OI-Wiki 。

「SPOJ 5971」LCMSUM

求值（多组数据）

$$\sum _{i=1}^n\mathrm{lcm}(i,n)\text{s.t.}1⩽T⩽3\times 10^5,1⩽n⩽10^6$$

易得原式即

$$\sum _{i=1}^n\frac{i\cdot n}{\gcd (i,n)}$$

将原式复制一份并且颠倒顺序，然后将 n 一项单独提出，可得

$$\frac{1}{2}\cdot \left(\sum _{i=1}^{n-1}\frac{i\cdot n}{\gcd (i,n)}+\sum _{i=n-1}^1\frac{i\cdot n}{\gcd (i,n)}\right)+n$$

根据 $\gcd (i,n)=\gcd (n-i,n)$，可将原式化为

$$\frac{1}{2}\cdot \left(\sum _{i=1}^{n-1}\frac{i\cdot n}{\gcd (i,n)}+\sum _{i=n-1}^1\frac{i\cdot n}{\gcd (n-i,n)}\right)+n$$

两个求和式中分母相同的项可以合并。

$$\frac{1}{2}\cdot \sum _{i=1}^{n-1}\frac{n^2}{\gcd (i,n)}+n$$

即

$$\frac{1}{2}\cdot \sum _{i=1}^n\frac{n^2}{\gcd (i,n)}+\frac{n}{2}$$

可以将相同的 $\gcd (i,n)$ 合并在一起计算，故只需要统计 $\gcd (i,n)=d$ 的个数。当 $\gcd (i,n)=d$ 时，$\gcd (\frac{i}{d},\frac{n}{d})=1$，所以 $\gcd (i,n)=d$ 的个数有 $\phi (\frac{n}{d})$ 个。

故答案为

$$\frac{1}{2}\cdot \sum _{d\mid n}\frac{n^2\cdot \phi (\frac{n}{d})}{d}+\frac{n}{2}$$

变换求和顺序，设 $d^{\prime }=\frac{n}{d}$，合并公因式，式子化为

$$\frac{1}{2}n\cdot \left(\sum _{d^{\prime }\mid n}{d}^{\prime }\cdot \phi (d^{\prime })+1\right)$$

设 $g(n)=\sum _{d\mid n}d\cdot \phi (d)$，已知 $g$ 为积性函数，于是可以 $\Theta (n)$ 筛出。每次询问 $\Theta (1)$ 计算即可。

下面给出这个函数筛法的推导过程：

首先考虑 $g(p_j^k)$ 的值，显然它的约数只有 $p_j^0,p_j^1,\cdots ,p_j^k$，因此

$$g(p_j^k)=\sum _{w=0}^k{p}_j^w\cdot \phi (p_j^w)$$

又有 $\phi (p_j^w)=p_j^{w-1}\cdot (p_j-1)$，则原式可化为

$$\sum _{w=0}^k{p}_j^{2w-1}\cdot (p_j-1)$$

于是有

$$g(p_j^{k+1})=g(p_j^k)+p_j^{2k+1}\cdot (p_j-1)$$

那么，对于线性筛中的 $g(i\cdot p_j)(p_j|i)$，令 $i=a\cdot p_j^w(\gcd (a,p_j)=1)$，可得

$$g(i\cdot p_j)=g(a)\cdot g(p_j^{w+1})$$

$$g(i)=g(a)\cdot g(p_j^w)$$

即

$$g(i\cdot p_j)-g(i)=g(a)\cdot p_j^{2w+1}\cdot (p_j-1)$$

同理有

$$g(i)-g(\frac{i}{p_j})=g(a)\cdot p_j^{2w-1}\cdot (p_j-1)$$

因此

$$g(i\cdot p_j)=g(i)+\left(g(i)-g(\frac{i}{p_j})\right)\cdot p_j^2$$

时间复杂度：$\Theta (n+T)$

代码

#include <cstdio>
const int N = 1000000;
int tot, p[N + 5];
long long g[N + 5];
bool flg[N + 5];  //标记数组

void solve() {
  g[1] = 1;
  for (int i = 2; i <= N; ++i) {
    if (!flg[i]) {
      p[++tot] = i;
      g[i] = (long long)1 * i * (i - 1) + 1;
    }
    for (int j = 1; j <= tot && i * p[j] <= N; ++j) {
      flg[i * p[j]] = 1;
      if (i % p[j] == 0) {
        g[i * p[j]] =
            g[i] + (g[i] - g[i / p[j]]) * p[j] * p[j];  //代入推出来的式子
        break;
      }
      g[i * p[j]] = g[i] * g[p[j]];
    }
  }
}
int main() {
  int T, n;
  solve();  //预处理g数组
  scanf("%d", &T);
  for (int i = 1; i <= T; ++i) {
    scanf("%d", &n);
    printf("%lld\n", (g[n] + 1) * n / 2);
  }
  return 0;
}