《GMOJ-Senior-4496 【GDSOI 2016 第一题】互补约数》题解

题目大意

给出一个正整数$n$，求${\sum }_{i=1}^n{\sum }_{j|i}\gcd (j,\frac{i}{j})$。
对于$20\%$的数据，$n\le 10^5$；
对于$50\%$的数据，$n\le 10^7$；
对于$80\%$的数据，$n\le 10^{10}$；
对于$100\%$的数据，$n\le 10^{11}$。

分析

这是一道数论题。题目中给出的式子不好直接算，于是我们要化简它。

化简式子

令$k=\frac{i}{j}$，并先枚举$k$，再枚举$j$，则$$原式=\sum _{k=1}\sum _{j=1}\gcd (j,k)[jk\le n]$$
因为要满足$jk\le n$，所以$k\le n$，$j\le \lfloor \frac{n}{k}\rfloor$，即$$原式=\sum _{k=1}^n\sum _{j=1}^{\lfloor \frac{n}{k}\rfloor }\gcd (j,k)$$
考虑枚举$\gcd$函数的值，因为对于乘积小于等于$n$的两正整数的最大公约数小于等于$\sqrt{n}$，所以有$$原式=\sum _{i=1}^{\lfloor \sqrt{n}\rfloor }i\sum _{k=1}^n\sum _{j=1}^{\lfloor \frac{n}{k}\rfloor }[\gcd (j,k)=i]$$
令$j^{\prime }=\frac{j}{i}$，$k^{\prime }=\frac{k}{i}$，则$$原式=\sum _{i=1}^{\lfloor \sqrt{n}\rfloor }i\sum _{k^{\prime }=1}^{\lfloor \frac{n}{i}\rfloor }\sum _{j^{\prime }=1}^{\lfloor \frac{n}{i^2{k}^{\prime }}\rfloor }[\gcd (j^{\prime },k^{\prime })=1]$$
因为${\sum }_{d|n}\mu (d)=[n=1]$，所以$$原式=\sum _{i=1}^{\lfloor \sqrt{n}\rfloor }i\sum _{k^{\prime }=1}^{\lfloor \frac{n}{i}\rfloor }\sum _{j^{\prime }=1}^{\lfloor \frac{n}{i^2{k}^{\prime }}\rfloor }\sum _{d|\gcd (j^{\prime },k^{\prime })}\mu (d)$$
改为依次枚举$d,j^{\prime },k^{\prime }$，有$$原式=\sum _{i=1}^{\lfloor \sqrt{n}\rfloor }i\sum _{d=1}\mu (d)\sum _{d|j^{\prime }}^{\lfloor \frac{n}{i}\rfloor }\sum _{d|k^{\prime }}^{\lfloor \frac{n}{i^2{j}^{\prime }}\rfloor }1=\sum _{i=1}^{\lfloor \sqrt{n}\rfloor }i\sum _{d=1}\mu (d)\sum _{d|j^{\prime }}^{\lfloor \frac{n}{i}\rfloor }\lfloor \frac{n}{i^2{j}^{\prime }d}\rfloor$$
令$a=\frac{j^{\prime }}{d}$，则有$$原式=\sum _{i=1}^{\lfloor \sqrt{n}\rfloor }i\sum _{d=1}\mu (d)\sum _{a=1}^{\lfloor \frac{n}{di}\rfloor }\lfloor \frac{n}{a{d}^2{i}^2}\rfloor$$
令$T=di$，并改为依次枚举$i,T$，则$$原式=\sum _{i=1}^{\lfloor \sqrt{n}\rfloor }i\sum _{i|T}\mu (\frac{T}{i})\sum _{a=1}^{\lfloor \frac{n}{T}\rfloor }\lfloor \frac{n}{T^{2}a}\rfloor$$
改变$i,T$的枚举顺序，则$$原式=\sum _{T=1}\sum _{i|T}\mu (\frac{T}{i})i\sum _{a=1}^{\lfloor \frac{n}{T}\rfloor }\lfloor \frac{n}{T^{2}a}\rfloor$$
因为${\sum }_{d|n}\mu (\frac{n}{d})d=\varphi (n)$，所以$$原式=\sum _{T=1}\varphi (T)\sum _{a=1}^{\lfloor \frac{n}{T}\rfloor }\lfloor \frac{n}{T^{2}a}\rfloor$$
因为当$\lfloor \frac{n}{T^{2}a}\rfloor \ge 1$时$a\le \lfloor \frac{n}{T^2}\rfloor$，且$\lfloor \frac{a}{bc}\rfloor =\lfloor \frac{\lfloor \frac{a}{b}\rfloor }{c}\rfloor$，所以$$原式=\sum _{T=1}\varphi (T)\sum _{a=1}^{\lfloor \frac{n}{T^2}\rfloor }\lfloor \frac{n}{T^{2}a}\rfloor =\sum _{T=1}\varphi (T)\sum _{a=1}^{\lfloor \frac{n}{T^2}\rfloor }\lfloor \frac{\lfloor \frac{n}{T^2}\rfloor }{a}\rfloor$$
因为当$\lfloor \frac{n}{T^2}\rfloor \ge 1$时$T\le \lfloor \sqrt{n}\rfloor$，所以
$$原式=\sum _{T=1}^{\lfloor \sqrt{n}\rfloor }\varphi (T)\sum _{a=1}^{\lfloor \frac{n}{T^2}\rfloor }\lfloor \frac{\lfloor \frac{n}{T^2}\rfloor }{a}\rfloor$$
我们发现，现在的式子可以被拆分为$\varphi (T)$和${\sum }_{a=1}^{\lfloor \frac{n}{T^2}\rfloor }\lfloor \frac{\lfloor \frac{n}{T^2}\rfloor }{a}\rfloor$两部分计算。

计算$\varphi (T)$

我们发现我们要算出$1$到$\lfloor \sqrt{n}\rfloor$的所有整数的$\varphi$函数的值。注意到对于正整数$n$和素数$p$，有$\varphi (p)=p-1$，且当$p\nmid n$时$\varphi (np)=\varphi (n)\varphi (p)=(p-1)\varphi (n)$、当$p|n$时$\varphi (np)=p\varphi (n)$，于是我们考虑在一个线性筛法的基础上求出所有$\varphi$函数的值。我们在添加新素数时更新素数的函数值，并在筛数时根据情况讨论被筛数的函数值，这样就可以得出$1$到$\lfloor \sqrt{n}\rfloor$的所有整数的$\varphi$函数的值了。

计算${\sum }_{a=1}^{\lfloor \frac{n}{T^2}\rfloor }\lfloor \frac{\lfloor \frac{n}{T^2}\rfloor }{a}\rfloor$

可以发现，对于$\lfloor \frac{\lfloor \frac{n}{T^2}\rfloor }{a}\rfloor$，当式子中$n,T$一定、$a$逐渐增大时，式子的值可以被划分成一段段值相等的区间，且当$a$增大到一定大小时，这种区间的大小会变得很可观。于是我们考虑从这个方向优化计算。我们考虑计算出每个区间的左右端点$l,r$，并用一个变量累加每个区间里的总和以求出${\sum }_{a=1}^{\lfloor \frac{n}{T^2}\rfloor }\lfloor \frac{\lfloor \frac{n}{T^2}\rfloor }{a}\rfloor$。可以发现，$r=\lfloor \frac{n}{\lfloor \frac{n}{l}\rfloor }\rfloor =\lfloor \frac{n}{区间内数字的值}\rfloor$，且$下一个区间的左端点=当前区间的右端点+1$，这样我们就能快速计算出每个区间的左右端点了。
此外，我们发现对于$\lfloor \frac{n}{T^2}\rfloor$（以及${\sum }_{a=1}^{\lfloor \frac{n}{T^2}\rfloor }\lfloor \frac{\lfloor \frac{n}{T^2}\rfloor }{a}\rfloor$），当$n$一定、$T$逐渐增大时也有类似的性质，于是我们也可以用类似的方法优化。

代码

根据思路，可以写出如下代码：

#include<cstdio> //定义头文件
#include<cmath>
using ll=long long; //将ll类型定义为long long类型
bool book[316228]; //筛法用的标记数组
ll prime[27294] /*筛法得出的素数*/ ,phi[316228]={0,1} /*phi函数的值*/ ;
int main()
{
	freopen("gcd.in","r",stdin); //定义文件输入输出
	freopen("gcd.out","w",stdout);
	ll n;
	scanf("%lld",&n); //读入n
	const ll sn=sqrt(n); //预处理floor(sqrt(n))
	for(ll i=2;i<=sn;++i) //通过筛法计算phi函数的值
	{
		if(!book[i])
		{
			prime[++prime[0]]=i; //增加一个素数
			phi[i]=i-1;
		}
		for(ll j=1,max_p=sn/i;j<=prime[0]&&prime[j]<=max_p;++j) //筛数
		{
			book[i*prime[j]]=1;
			if(i%prime[j]==0)
			{
				phi[i*prime[j]]=phi[i]*prime[j];
				break;
			}
			phi[i*prime[j]]=phi[i]*phi[prime[j]];
		}
	}
	for(ll i=2;i<=sn;++i) //求phi函数的前缀和，为下面准备
	{
		phi[i]+=phi[i-1];
	}
	ll ans=0; //统计答案
	for(ll l=1 /*区间左端点*/ ,val=n/(l*l) /*区间内数的值*/ ,r /*区间右端点*/ ;l<=sn&&val>0;l=r+1,val=n/(l*l)) //利用floor(n/T^2)的性质计算ans
	{
		r=sqrt(n/val); //计算右端点
		if(r>sn)
		{
			r=sn;
		}
		ll t=0; //统计sum(floor(floor(n/T^2)/a))的值
		for(ll i=1 /*区间左端点*/ ,v=val/i /*区间内数的值*/ ,j /*区间右端点*/ ;i<=val&&v>0;i=j+1,v=val/i) //利用floor(floor(n/T^2)/a)的性质计算t
		{
			j=val/v; //计算右端点
			if(j>val)
			{
				j=val;
			}
			t+=(j-i+1)*v; //更新t
		}
		ans+=(phi[r]-phi[l-1])*t; //更新ans
	}
	printf("%lld",ans); //输出ans
	return 0;
}

总结

这是一道比较好的数论题，考察了多个数论技巧。