JZOJ 5516 Function

Function

Description

$σ_0(n)$=$\sum_{d|n}1$
求

$$\sum_{i=1}^n\sum_{d|i}\mu (d)\sigma_0({i\over d})\sigma_0({i\over d})$$

Data Constraint

$n$<=$10^9$

Solution

$σ_0(n)$就是$n$的约数个数。
显然有$σ_0=1*1$。

设$g(n)$=$σ_0^2(n)$,$f(n)$=$\sum_{d|i}\mu (d)g({i\over d})$，则有$f=\mu *g$
显然$\mu$和$g$都是积性函数，所以$f$也是积性函数。
题目就是让我们求$\sum_{i=1}^nf(i)$。

易得$f(p^k)$=$2k+1$（其中$p$为质数）
设$i=p_1^{a_1}*p_2^{a_2}*...*p_w^{a_w}$
由$f$是积性函数可得$f(i)$=$\Pi_{i=1}^w(2a_i+1)$=$\sigma_0(i^2)$
所以$Answer$=$\sum_{i=1}^n\sigma_0(i^2)$

我们有$\sigma_0(i^2)$=$\Pi_{i=1}^w(2a_i+1)$
将右边的式子拆开，然后考虑每一项的贡献，便可得

$$\sigma_0(i^2)=\sum_{C\subseteq\{1,2,3...,w\}}{2^{|C|}}*\Pi a_i({i\subseteq C})$$
经过仔细的思索便可以将上式写成另一种形式
$$\sigma_0(i^2)=\sum_{d|i}2^{r(d)}$$
其中 ${r(d)}$表示 $d$的不同质因子的个数。
设$h(x)$= $2^{r(x)}$,则有 $f=h*1$

考虑$h(x)$的另一个意义即$x$的无平方因子的约数个数，即

$$h(x)=\sum_{d|x}\mu^2(d)$$
再设$t(x)=\mu^2(x)$,则有$h=t*1$。

所以

$$f=h*1=(t*1)*1=t*(1*1)=t*σ_0$$

所以

$$f(i)=\sum_{d|i}\mu^2(d)*\sigma_0({i\over d})$$

$$Answer=\sum_{i=1}^n\sum_{d|i}\mu^2(d)*\sigma_0({i\over d})=\sum_{i=1}^n\mu^2(i)*[\sum_{j=1}^{\lfloor {n\over i}\rfloor}\sigma_0(j)]$$

对式子分块求解，预处理出$\mu^2(i)$和$\sigma_0(i)$前$n^{2\over 3}$的前缀和，若超出了这个范围就$O(\sqrt N)$暴力求解，听说复杂度和杜教筛的复杂度一样，为$O(n^{2\over 3})$。

求$\sum_{i=1}^n\sigma_0(i)$=$\sum_{i=1}^n\lfloor {n\over i} \rfloor$，分块$O(\sqrt n)$求解即可。

求$\sum_{i=1}^n\mu^2(i)$，考虑从意义出发，这个式子就是让我们求$1$~$n$内无平方因子的数的个数，考虑用莫比乌斯函数进行容斥，可得

$$\sum_{i=1}^n\mu^2(i)=\sum_{i=1}^{\sqrt n}\mu(i)\lfloor {n\over i^2}\rfloor$$

$O(\sqrt n)$暴力统计即可。

Code

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<cmath>

#define fo(i,j,l) for(int i=j;i<=l;++i)
#define fd(i,j,l) for(int i=j;i>=l;--i)

using namespace std;
typedef long long ll;
const ll K=12e5,N=15e5;
int T,oo;
int mu[N],ss[N],cs[N];
ll qz[N],g[N],q[N],n;
bool bz[N];

inline ll min(ll a,ll b)
{return a<b?a:b;}

inline ll findmu(ll p)
{
    if(p<=K)return qz[p];
    ll u=(ll)sqrt(p),ans=0;
    fo(i,1,u)ans=ans+mu[i]*((p/i)/i);
    return ans;
}

inline ll findysh(ll p)
{
    if(p<=K)return q[p]; 
    int l=1,r; ll ans=0;
    while(l<=p){
        r=p/(p/l);
        ans=ans+(p/l)*(r-l+1);
        l=r+1;
    }
    return ans;
}

int main()
{
    freopen("function.in","r",stdin);
    freopen("function.out","w",stdout);
    cin>>T;
    cs[1]=mu[1]=g[1]=1;
    fo(i,2,K){
        if(!bz[i])ss[++oo]=i,mu[i]=-1,g[i]=2,cs[i]=2;
        fo(j,1,oo){
            if(i*ss[j]>K)break;
            bz[i*ss[j]]=true;
            if(i%ss[j]==0){
                mu[i*ss[j]]=0;
                cs[i*ss[j]]=cs[i]+1;
                g[i*ss[j]]=(g[i]/cs[i])*(cs[i]+1);
                break;
            }else cs[i*ss[j]]=2,g[i*ss[j]]=g[i]*2,mu[i*ss[j]]=-mu[i];
        }
    }
    fo(i,1,K)qz[i]=qz[i-1]+(mu[i]!=0);
    fo(i,1,K)q[i]=q[i-1]+g[i];
    fo(tt,1,T){
        scanf("%lld",&n);
        ll ans=0;
        ll l=1,r;
        while(l<=n){
            r=n/(n/l);
            ans=ans+(findmu(r)-findmu(l-1))*findysh(n/l);
            l=r+1;
        }
        printf("%lld\n",ans);
    }
}