BZOJ4176 Lucas的数论

题目大意

去年的Lucas非常喜欢数论题，但是一年以后的Lucas却不那么喜欢了。

在整理以前的试题时，发现了这样一道题目“求$\sum _{i=1}^{n}f(i)$”，其中$f(i)$表示$i$的约数个数。他现在长大了，题目也变难了。

给定$n$，求如下表达式的值：

$$\sum _{i=1}^n\sum _{j=1}^{n}f(ij)$$

输出答案模$10^9+7$后的值。

$1\le n\le 10^9$

题解

前置知识：

• 莫比乌斯反演
• 数论分块
• 杜教筛

有一道类似的题，可以看一下：【SDOI2015】约数个数和。

式子的证明

首先，我们要知道：

$$d(ij)=\sum _{x|i}\sum _{y|j}[\gcd (x,y)=1]$$

证明：

如果$ij$的因子$k$中有因子$p^c$，$i$中有因子$p^a$，$j$中有因子$p^b$，其中$p$为质数。那么我们规定

• 当$c\le a$时，这个因子$p$都在$a$中选择
• 当$c>a$时，在$i$中选$p^a$，在$j$中选$p^{c-a}$

对于$ij$的因子$k$的每个因子都这样处理。

如果令$x,y$互质，则

• 若$x,y$中都没有因子$p$，则
• 若$x$中有因子$p$且次数为$k$，则相当于在$x$中选了$p^k$
• 若$y$中有因子$p$且次数为$k$，则相当于在$x$中选了$p^a$，在$j$中选了$p^k$

$ij$的各个因子$k$对应各个互质的$x,y$，所以等式成立。

推式子

根据上面的式子，有

$$\sum _{i=1}^n\sum _{j=1}^{n}d(ij)=\sum _{i=1}^n\sum _{j=1}^n\sum _{x|i}\sum _{y|j}[\gcd (x,y)=1]$$

由莫比乌斯函数的性质，原式可变为

$$\sum _{i=1}^n\sum _{j=1}^n\sum _{x|i}\sum _{y|j}\sum _{k|i,k|j}\mu (k)$$

先枚举$k$，则转换为

$$\sum _{k=1}^n\mu (k)\sum _{i=1}^{\lfloor \frac{n}{k}\rfloor }\sum _{i=1}^{\lfloor \frac{n}{k}\rfloor }\lfloor \frac{n}{ki}\rfloor \lfloor \frac{n}{kj}\rfloor$$

也就是

$$\sum _{k=1}^n\mu (k)\times (\sum _{i=1}^{\lfloor \frac{n}{k}\rfloor }\lfloor \frac{n}{ki}\rfloor {)}^2$$

令$g(n)=\sum _{i=1}^n\lfloor \frac{n}{i}\rfloor$，则$g(n)=\sum _{i=1}^{n}f(i)$。

为什么呢？因为对于每一个$i$，加上$\lfloor \frac{n}{i}\rfloor$就相当于对$i,2i,3i,\dots$都算一次贡献，也就是让所有有约数为$i$的数计算一次约数为$i$的贡献。而$1$到$n$的每个数都有被计算贡献，所以$g(n)=\sum _{i=1}^{n}f(i)$。

用杜教筛求$\mu (i)$和$f(i)$的值（$f(i)$是约数个数，是积性函数，所以可以用杜教筛来求）。

时间复杂度为$O(n^{\frac{2}{3}})$。

code

#include<bits/stdc++.h>
using namespace std;
const int N=2000000;
const long long mod=1000000007;
int z[N+5],p[N+5],mu[N+5],y[N+5];
long long ans=0,smu[N+5],sy[N+5];
map<long long,long long>mmu,my;
void init(){
	mu[1]=1;
	for(int i=2;i<=N;i++){
		if(!z[i]){
			p[++p[0]]=i;mu[i]=-1;
		}
		for(int j=1;j<=p[0]&&i*p[j]<=N;j++){
			z[i*p[j]]=1;
			if(i%p[j]==0){
				mu[i*p[j]]=0;
				break;
			}
			mu[i*p[j]]=-mu[i];
		}
	}
	for(int i=1;i<=N;i++){
		smu[i]=smu[i-1]+mu[i];
		for(int j=i;j<=N;j+=i) ++y[j];
		sy[i]=(sy[i-1]+y[i])%mod;
	}
}
long long gtmu(int t){
	if(t<=N) return smu[t];
	if(mmu.count(t)) return mmu[t];
	long long re=0;
	for(int l=2,r;l<=t;l=r+1){
		r=t/(t/l);
		re=(re+(r-l+1)*gtmu(t/l)%mod)%mod;
	}
	return mmu[t]=(1-re+mod)%mod;
}
long long gty(int t){
	if(t<=N) return sy[t];
	if(my.count(t)) return my[t];
	long long re=0;
	for(int l=2,r;l<=t;l=r+1){
		r=t/(t/l);
		re=(re+(gtmu(r)-gtmu(l-1)+mod)%mod*gty(t/l)%mod)%mod;
	}
	return my[t]=(t-re+mod)%mod;
}
int main()
{
	int n;
	init();
	scanf("%d",&n);
	for(int l=1,r;l<=n;l=r+1){
		r=n/(n/l);
		long long t=gty(n/l);
		ans=(ans+(gtmu(r)-gtmu(l-1)+mod)%mod*t%mod*t%mod)%mod;
	}
	printf("%lld",ans);
	return 0;
}