[2018雅礼集训1-10]Function 积性函数前缀和

题面
题目要我们求

$$\sum_{i=1}^n\sum_{d|i}\mu(d)*\sigma_0^2(\frac{i}{d})$$
设 $f(i)=\sum_{d|i}\mu(d)*\sigma_0^2(\frac{i}{d})$，感觉 $f$会是个很优美的东西，打表发现
$$f(i)=\sigma_0(i^2)$$
怎么证明呢？考虑 $\sigma_0(i^2)=\sum_{d|i}\mu(d)*\sigma_0^2(\frac{i}{d})$用莫比乌斯反演反过来就是 $\sigma_0^2(n)=\sum_{d|n}\sigma_0(n^2)$。这里给出一种推式子的证法：
$$\sum_{d|n}\sigma_0(d^2)=\sum_{d|n}\sum_{i|d}\sum_{j|d}[(i,j)=1]=\sum_{d|n}\sum_{i|d}\sum_{j|d}[(i*\lfloor\frac{n}{d}\rfloor,j*\lfloor\frac{n}{d}\rfloor)=\lfloor\frac{n}{d}\rfloor]$$
$$=\sum_{d|n}\sum_{i|n}\sum_{j|n}[(i,j)=\lfloor\frac{n}{d}\rfloor]=\sum_{d|n}\sum_{i|n}\sum_{j|n}[(i,j)=d]$$
其中第一步利用了 $\sigma_0(n^2)$等于 $n$的有序互质约数对个数，大概考虑$n$的某一个质因数项 $p_i^{k_i}$，前者的贡献是 $2*k_i+1$，后者是 $(p^i,p^j)(i=0||j=0)$的对数，也是 $2*k_i+1$。
$$\sigma_0^2(n)=\sum_{i|n}\sum_{j|n}1=\sum_{d|n}\sum_{i|n}\sum_{j|n}[(i,j)=d]$$
得证。
然后就和 SPOJ DIVCNT2一样了。
代码：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#define ll long long
using namespace std;
const int maxn=1000000;
int n,pri[maxn+5],num,mu[maxn+5],d[maxn+5],sm[maxn+5],sd[maxn+5];
bool flag[maxn+5];
void getpri(int n)
{
    memset(flag,1,sizeof(flag));
    flag[1]=0;mu[1]=1;
    for(int i=2;i<=n;i++)
    {
        if(flag[i]) pri[++num]=i,mu[i]=-1;
        for(int j=1;j<=num&&i*pri[j]<=n;j++)
        {
            flag[i*pri[j]]=0;
            if(i%pri[j]==0) {mu[i*pri[j]]=0;break;}
            mu[i*pri[j]]=-mu[i];
        }
    }
}
ll qsd(int m)
{
    if(m<=maxn) return sd[m];
    ll re=0;
    for(int l=1,r=1,k;l<=m;l=r+1)
    {
        k=m/l;r=m/k;
        re+=(ll)(r-l+1)*k;
    }
    return re;
}
ll qsm(int m)
{
    if(m<=maxn) return sm[m];
    int gen=round(sqrt(m)+1);ll re=0;
    for(int i=1;i<=gen;i++)
        re+=mu[i]*(m/(i*i));    
    return re;  
}
ll solve()
{
    ll re=0;
    for(int l=1,r=1,k;l<=n;l=r+1)
    {
        k=n/l;r=n/k; 
        re+=qsd(k)*(qsm(r)-qsm(l-1));
    }
    return re;
}
int main()
{
    int ca;
    scanf("%d",&ca);
    getpri(maxn);
    for(int i=1;i<=maxn;i++)
        for(int j=1;i*j<=maxn;j++)
            d[i*j]++;       
    for(int i=1;i<=maxn;i++)
        sm[i]=sm[i-1]+mu[i]*mu[i],sd[i]=sd[i-1]+d[i];
    while(ca--)
    {
        scanf("%d",&n);
        printf("%lld\n",solve());
    }
    return 0;
}