[BZOJ2226][Spoj 5971] LCMSum（莫比乌斯反演）

题目描述

传送门

题解

画一波柿子

$$\sum\limits_{i=1}^n[i,j]$$
$$=\sum\limits_{i=1}^n{ni\over (i,j)}$$
$$=n\sum\limits_{i=1}^n\sum\limits_{d=1}^n[(i,n)=d]{i\over d}$$
令 $i=id$
$$=n\sum\limits_{d|n}\sum\limits_{i=1}^{n\over d}[(i,{n\over d})=1]i$$
利用反演公式 $[n=1]=\sum\limits_{d|n}\mu(d)$
$$=n\sum\limits_{d|n}\sum\limits_{i=1}^{n\over d}i\sum\limits_{t|(i,{n\over d})}\mu(t)$$
$$=n\sum\limits_{d|n}\sum\limits_{t|{n\over d}}\sum\limits_{i=1}^{n\over d}[t|i]i\mu(t)$$
令 $s(n)=\sum\limits_{i=1}^ni={i(i+1)\over 2}$
$$n\sum\limits_{d|n}\sum\limits_{t|{n\over d}}s({{n\over d}\over t})t\mu(t)$$
设 $f(n)=\sum\limits_{d|n}s({n\over d})t\mu(t)$
$$=n\sum\limits_{d|n}f({n\over d})$$
令 $F(n)=\sum\limits_{d|n}f({n\over d})$
那么答案就是 $nF(n)$了嘛…
可以发现 $f$和$F$都可以用埃式筛法筛出来
那么询问就是 $O(1)$的啦…
时间复杂度 $O(nloglogn+T)$

需要注意的是，这个公式：$(i,j)=d->({i\over d},{j\over d})=1$条件是$d|i$且$d|j$

代码

#include<algorithm>
#include<iostream>
#include<cstring>
#include<cstdio>
#include<cmath>
using namespace std;
#define N 1000005
#define LL long long

int T,n;
LL ans;
int p[N],prime[N],mu[N];
LL f[N],F[N];

void get(int n)
{
    mu[1]=1;
    for (int i=2;i<=n;++i)
    {
        if (!p[i])
        {
            prime[++prime[0]]=i;
            mu[i]=-1;
        }
        for (int j=1;j<=prime[0]&&i*prime[j]<=n;++j)
        {
            p[i*prime[j]]=1;
            if (i%prime[j]==0)
            {
                mu[i*prime[j]]=0;
                break;
            }
            else mu[i*prime[j]]=-mu[i];
        }
    }
    for (int i=1;i<=n;++i)
        for (int j=i;j<=n;j+=i)
            f[j]+=(LL)mu[i]*i*(j/i)*(j/i+1)/2;
    for (int i=1;i<=n;++i)
        for (int j=i;j<=n;j+=i)
            F[j]+=f[j/i];
}
int main()
{
    get(1000000);
    scanf("%d",&T);
    while (T--)
    {
        scanf("%d",&n);
        ans=(LL)F[n]*n;
        printf("%lld\n",ans);
    }
}