【JZOJ 4496】【GDSOI 2016】第一题 互补约数 （两种解法）

Description

求

$$\sum_{i=1}^n\sum_{d|i}gcd(d,\dfrac id)$$
对于100%的数据， $n\leq 10^{11}$。

Analysis

方法一

这是一个不需要莫比乌斯反演的方法。

$$\sum_{i=1}^n\sum_{d|i}gcd(d,\dfrac id)$$
$$=\sum_{i*j\leq n}gcd(i,j)$$
$$=\sum_{i*j\leq n}\sum_{d|gcd(i,j)}\varphi(d)$$
这里用到了一个 $\varphi$的性质， $\sum_{d|n}\varphi(d)=n$，具体证明可以看 我的博客。
$$=\sum_{i*j\leq n}\sum_{d|i\bigcap d|j}\varphi(d)$$
调换主体，得
$$=\sum_{d=1}^{\sqrt n}\varphi(d)\sum_{i*j\leq\lfloor\dfrac{n}{d^2}\rfloor}1$$
$$=\sum_{d=1}^{\sqrt n}\varphi(d)\sum_{i=1}^{\lfloor\dfrac{n}{d^2}\rfloor}\lfloor\dfrac{n}{d^2i}\rfloor$$
随着 $d$的增长，$d^2$增长迅速，因此 $\lfloor\dfrac{n}{d^2}\rfloor$会迅速收敛到很小，并且有许多的 $\lfloor\dfrac{n}{d^2i}\rfloor$的值是一样的，所以可以分块处理。时间复杂度是很优的。

方法二

这次我们用莫比乌斯反演。

$$\sum_{i=1}^n\sum_{d|i}gcd(d,\dfrac id)$$
$$=\sum_{i=1}^n\sum_{j=1}^{\lfloor\dfrac ni\rfloor}gcd(i,j)$$
设 $f(d)$表示上式中满足 $gcd(i,j)=d$的 $i,j$有多少对;
$g(d)$表示上式中满足 $d|gcd(i,j)$的 $i,j$有多少对。
首先有
$$g(d)=\sum_{i=1}^{\lfloor\dfrac nd\rfloor}f(d*i)$$
反演
$$f(d)=\sum_{i=1}^{\lfloor\dfrac nd\rfloor}g(d*i)*\mu(i)$$
而我们要求的 $$Ans=\sum_{d=1}^nd*f(d)$$
$$=\sum_{d=1}^nd\sum_{i=1}^{\lfloor\dfrac nd\rfloor}g(d*i)*\mu(i)$$
设 $T=d*i$，再调换主体
$$=\sum_{T=1}^ng(T)\sum_{i|T}\dfrac Ti*\mu(i)$$
但其实 $T$只用枚举到$\sqrt n$，因为 $\sqrt n$是在 $i*j\leq n$的范围内 $gcd(i,j)$的最大取值，此时 $i=j=\sqrt n$
那么 $g(T)$怎么求呢？
若 $d|gcd(i,j)$，则可以设 $dx,dy$分别为任意的 $i,j$，因为必有 $d|gcd(dx,dy)$
所以我们枚举 $x$，因为$dx*dy\leq n$，所以 $x$只需要枚举到$\lfloor\dfrac{n}{d^2}\rfloor$
$$g(T)=\sum_{x=1}^{\lfloor\dfrac{n}{T^2}\rfloor}\lfloor\dfrac{n}{T^2x}\rfloor$$
所以 $$Ans=\sum_{T=1}^{\sqrt n}\sum_{x=1}^{\lfloor\dfrac{n}{T^2}\rfloor}\lfloor\dfrac{n}{T^2x}\rfloor*\sum_{i|T}\dfrac Ti*\mu(i)$$
这个东西直接做，预处理一些东西，就可以AC了。
但我们的追求更远。
上面这个式子有没有一点眼熟？
我们把方法一最后得出的结论搬下来
$$Ans=\sum_{d=1}^{\sqrt n}\varphi(d)\sum_{i=1}^{\lfloor\dfrac{n}{d^2}\rfloor}\lfloor\dfrac{n}{d^2i}\rfloor$$ ，
两个式子比较一下，许多东西非常相似，所以我们可以提出这样一个猜想
$$\varphi(n)=\sum_{d|n}\dfrac nd*\mu(d)$$
证明如下：
设 $f(n)=\varphi(n),g(n)=n$
根据上面说的那个欧拉函数的性质，得
$$g(n)=n=\sum_{d|n}\varphi(d)=\sum_{d|n}f(d)$$
反演
$$f(n)=\varphi(n)=\sum_{d|n}g(\dfrac nd)*\mu(d)=\sum_{d|n}\dfrac nd*\mu(d)$$
得证！
殊途同归！

Code

#include<cstdio>
#include<cmath>
#include<algorithm>
#define fo(i,a,b) for(int i=a;i<=b;i++)
using namespace std;
typedef long long ll;
const int N=316230;
int phi[N],pri[N];
bool bz[N];
void pre(int n)
{
    phi[1]=1;
    fo(i,2,n)
    {
        if(!bz[i])
        {
            phi[i]=i-1;
            pri[++pri[0]]=i;
        }
        fo(j,1,pri[0])
        {
            int t=i*pri[j];
            if(t>n) break;
            bz[t]=1;
            if(i%pri[j]==0)
            {
                phi[t]=phi[i]*pri[j];
                break;
            }
            phi[t]=phi[i]*phi[pri[j]];
        }
    }
}
int main()
{
    freopen("gcd.in","r",stdin);
    freopen("gcd.out","w",stdout);
    ll n,ans=0;
    scanf("%lld",&n);
    int m=int(sqrt(n));
    pre(m);
    fo(p,1,m)
    {
        ll i,l=n/p/p,t=0;
        for(ll j=1;j<=l;j=i+1)
        {
            i=l/(l/j);
            t+=(i-j+1)*(l/j);
        }
        ans+=phi[p]*t;
    }
    printf("%lld",ans);
    fclose(stdin);fclose(stdout);
    return 0;
}