bzoj 2226: [Spoj 5971] LCMSum （反演）

题目描述

传送门

题目大意：

$$\sum_{i=1}^n lcm(i,n)$$

题解

又要化式子啦，开心。

$$\sum_{i=1}^n lcm(i,n)$$
$$\sum_{d=1}^n \sum_{i=1}^n {i*n\over d} [gcd(i,n)=d]$$
$$n\sum_{d|n} \sum_{i=1}^{n\over d} i*[gcd(i,{n\over d})=1]$$
设 $g(n)=\sum_{i=1}^n i*[gcd(i,n)=1]$ ,那么上式就是
$$n\sum_{d|n} g({n\over d})$$
考虑怎么求 $g(n)$
$$g(n)=\sum_{i=1}^n i*[gcd(i,n)=1]$$
$$\sum_{i=1}^n i \sum_{d|(i,n)} \mu(d)$$
$$\sum_{d|n} d*\mu(d) \sum_{i=1}^{n\over d} i$$
$$\sum_{d|n} d*\mu(d) (1+{n\over d}){n\over d}/2$$
$${n\over2} \sum_{d|n} (\mu(d)+{n\over d}*\mu(d))$$
$${n\over 2} \sum_{d|n} \mu(d) +\sum_{d|n} {n\over d}*\mu(d)$$
因为 $\sum_{d|n} \varphi(d)=id(n)$,根据莫比乌斯反演的公式 $F(n)=\sum_{d|n} f(d), f(n)=\sum_{d|n} F(d)*\mu({n\over d})$ ,所以 $\sum_{d|n} {n\over d}*\mu(d)=\varphi(n)$.
$\sum_{d|n}\mu(d)=[n=1]$,所以上面的式子在n>1时， $g(n)={n*\varphi(n)\over 2}$
这个与n互质数的和可以当做结论记住。
$$n\sum_{d|n} g({n\over d})$$ ,这个东西可以用艾氏筛法预处理，然后 $O(1)$查询。

代码

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cmath>
#include<cstring>
#define N 1000000
#define LL long long
using namespace std;
int T,n,pd[N+3],prime[N+3],f[N+3],phi[N+3];
LL g[N+3],ans[N+3];
void init()
{
    for (int i=2;i<=N;i++) {
        if (!pd[i]) {
            prime[++prime[0]]=i;
            phi[i]=i-1;
        }
        for (int j=1;j<=prime[0];j++) {
            if (i*prime[j]>N) break;
            pd[i*prime[j]]=1;
            if (i%prime[j]==0) {
                phi[i*prime[j]]=phi[i]*prime[j];
                break;
            }
            phi[i*prime[j]]=phi[i]*(prime[j]-1);
        }
    }
    g[1]=1;
    for (int i=2;i<=N;i++) g[i]=(LL)i*phi[i]/2;
    for (int i=1;i<=N;i++) 
        for (int j=i;j<=N;j+=i) ans[j]+=g[i];
}
int main()
{
    freopen("a.in","r",stdin);
    scanf("%d",&T);
    init();
    while (T--) {
        scanf("%d",&n);
        printf("%lld\n",n*ans[n]);
    }
}