GDSOI 2016 T1 互补约数

Description

求

$$\sum_{i=1}^{n}\sum_{d|i}gcd(d,{i\over d})$$
n<=10^11

Solution

首先，我们发现gcd中的两个东西是所有乘积不超过n的数对，即

$$Ans=\sum_{i}\sum_{j,i*j<=n}gcd(i,j)$$
然后 $$Ans=\sum_{i=1}^{n}\sum_{j=1}^{\lfloor{n\over i}\rfloor}gcd(i,j)$$
咦？
长得那么像莫比乌斯的经典形式。只不过j的上界是会变化的。
那么我们可以尝试一下。
设 $F(d)$表示上式中 $gcd(i,j)=d$的个数， $G(d)$表示 $d|gcd(i,j)$的个数。
根据反演， $$F(d)=\sum_{i=1}^{\lfloor{n\over d}\rfloor}G(id)\mu(i)$$
那么 $$Ans=\sum_{d=1}^{n}F(d)*d$$
展开 $$Ans=\sum_{d=1}^{n}d\sum_{i=1}^{\lfloor{n\over d}\rfloor}G(id)\mu(i)$$
考虑枚举 $T=i*d$
$$Ans=\sum_{T=1}^{n}G(T)\sum_{i|T}{T\over i}\mu(i)$$
似乎复杂度一点都没有优化？
我们考虑一下如何求G
如果两个数的gcd=T，那么这两个数可以表示成XT,YT.
那么我们可以枚举X, $$G(T)=\sum_{x=1}^{\lfloor{n\over {T^2}}\rfloor}\lfloor{n\over {T^2x}}\rfloor$$
这个东西的复杂度大概是 $O(N^{1\over 2})$
由于n的下面有个平方，于是这个东西的复杂度肯定没问题。
于是发现原来的式子中，T只需要枚举到 $\sqrt{n}$就好了。
那么我们可以考虑预处理出 $\sum_{i|T}{T\over i}\mu(i)$设为 $a(T)$
那么原式就变成了优美的 $$Ans=\sum_{T=1}^{\sqrt{n}}G(T)a(T)$$
时间复杂度大约为 $O(N^{3\over 4})$
实际远远不到，O(跑得过)O(∩_∩)O~

Code

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