acm-(数论、莫比乌斯反演)SP20173 DIVCNT2 - Counting Divisors (square)

传送门
本题是让求这样一个式子：${\sum }_{\mathbf{i}=1}^{\mathbf{n}}\mathbf{d}({\mathbf{i}}^2)$。
先看$\mathbf{d}({\mathbf{n}}^2)$如何化解，根据基础数论学习笔记(下)第十部分6.13和14我们可以知道$\mathbf{d}({\mathbf{n}}^2)={\sum }_{{\mathbf{d}}^{\prime }\mid \mathbf{n}}{\sum }_{\mathbf{k}\mid {\mathbf{d}}^{\prime }}|\mu (\mathbf{k})|$。方便起见假设$\mathbf{f}(\mathbf{n})=|\mu (\mathbf{n})|$，于是有$\mathbf{d}({\mathbf{n}}^2)=((\mathbf{f}\ast \mathbf{I})\ast \mathbf{I})(\mathbf{n})=(\mathbf{f}\ast (\mathbf{I}\ast \mathbf{I}))(\mathbf{n})=(\mathbf{f}\ast \mathbf{d})(\mathbf{n})={\sum }_{{\mathbf{d}}^{\prime }\mid \mathbf{n}}\mathbf{f}({\mathbf{d}}^{\prime })\mathbf{d}(\frac{\mathbf{n}}{{\mathbf{d}}^{\prime }})$。
代入原式我们可以得到：
$\begin{array}{rl}\sum _{i=1}^{n}d(i^2)& =\sum _{i=1}^n\sum _{d^{\prime }\mid i}f(d^{\prime })d(\frac{i}{d^{\prime }})\\ & =\sum _{d^{\prime }=1}^{n}f(d^{\prime })\sum _{i=1}^{\lfloor \frac{n}{d^{\prime }}\rfloor }d(i)\end{array}$
考虑如何求解$\mathbf{f}(\mathbf{i})$前缀和，显然${\sum }_{\mathbf{i}=1}^{\mathbf{n}}\mathbf{f}(\mathbf{i})={\sum }_{\mathbf{i}=1}^{\mathbf{n}}|\mu (\mathbf{i})|$这个式子表示的实际意义是小于等于n的无平方因子数的个数。因此我们可以利用容斥原理写成${\sum }_{\mathbf{i}=1}^{\lfloor \sqrt{\mathbf{n}}\rfloor }\mu (\mathbf{i})\lfloor \frac{\mathbf{n}}{{\mathbf{i}}^2}\rfloor$，具体证明见基础数论学习笔记(下)6.15，这个式子可以暴力计算，复杂度是$\mathbf{O}(\sqrt{\mathbf{n}})$。
然后我们再考虑$\mathbf{d}(\mathbf{i})$的前缀和，显然有${\sum }_{\mathbf{i}=1}^{\mathbf{n}}\mathbf{d}(\mathbf{i})={\sum }_{\mathbf{i}=1}^{\mathbf{n}}{\sum }_{{\mathbf{d}}^{\prime }\mid \mathbf{i}}1={\sum }_{\mathbf{i}=1}^{\mathbf{n}}\lfloor \frac{\mathbf{n}}{\mathbf{i}}\rfloor$，这个式子也可以分块求，然后我们就能够分块求出${\sum }_{{\mathbf{d}}^{\prime }=1}^{\mathbf{n}}\mathbf{f}({\mathbf{d}}^{\prime }){\sum }_{\mathbf{i}=1}^{\lfloor \frac{\mathbf{n}}{{\mathbf{d}}^{\prime }}\rfloor }\mathbf{d}(\mathbf{i})$，也就是最终的答案，然后一个优化使提前预处理出${\mathbf{n}}^{\frac{2}{3}}$以内的$\mathbf{f}与\mathbf{d}$的前缀和，这样类比杜教筛的话总复杂度可以达到$\mathbf{O}({\mathbf{n}}^{\frac{2}{3}})$。

ul n,d[maxn];
int prim[maxn],tot,summu[maxn],N;
char mu[maxn];
bool flag[maxn];
void init(int n){
	mu[1]=1;
	d[1]=1;
	FOR(i,2,n+1){
		if(!flag[i])prim[tot++]=i,mu[i]=-1,d[i]=2,summu[i]=2;
		for(register int j = 0;j<tot && prim[j]*i<=n;++j){
			flag[i*prim[j]]=1;
			if(i%prim[j]==0){
				d[i*prim[j]]=d[i]+d[i]/summu[i];
				summu[i*prim[j]]=summu[i]+1;
				break;
			}
			mu[i*prim[j]]=-mu[i];
			d[i*prim[j]]=2*d[i];
			summu[i*prim[j]]=2;
		}
	}
	FOR(i,1,n+1)d[i]+=d[i-1],summu[i]=(summu[i-1]+abs(mu[i]));
}
ul calf(ul n){
	if(n<=N)return summu[n];
	register int limt=sqrt(n);
	register ul ans=0;
	for(register ll i=1;i<=limt;++i)ans+=mu[i]*(n/(i*i));
	return ans;
}
ul cald(ul n){
	if(n<=N)return d[n];
	register ul i=1,j,ans=0;
	while(i<=n){
		j=n/(n/i);
		ans+=(j-i+1)*(n/i);
		i=j+1;
	}
	return ans;
}
void solve(){
	ul i=1,j,ans=0,pre=0,cur;
	while(i<=n){
		j=n/(n/i);
		ans+=((cur=calf(j))-pre)*cald(n/i);
		i=j+1;
		pre=cur;
	}
	wrn(ans);
}
int main(){
	int t;
	ul mxn=0;
	vector<ul>a;
	rd(&t);
	while(t--){
		rd(&n);
		mxn=max(mxn,n);
		a.push_back(n);
	}
	N=pow(mxn,2.0/3.0);
	if(N<=10000)init(10000);else init(N);
	FOR(i,0,a.size()){
		n=a[i];
		solve();
	}
}