【BZOJ 2671】Calc

传送门

---

problem

给出 $n$，统计满足下面条件的数对 $(a,b)$ 的个数：

• $1\le a<b\le n$；
• $(a+b)|ab$。

数据范围：$n\le 2^{31}-1$。

---

solution

我们要算的是

$$ans=\sum _{a=1}^n\sum _{b=a+1}^n[(a+b)|ab]$$

考虑 $(a+b)|ab$ 什么情况下成立。设 $\gcd (a,b)=d$，$a=ud$，$b=vd$ 的话，就是：

$$(u+v)d|uv{d}^2\Rightarrow (u+v)|uvd$$

由于 $\gcd (u,v)=1$，所以 $\gcd (u+v,uv)=1$，那么就要满足：

$$(u+v)|d$$

所以我们要求的其实是：

$$ans=\sum _{u=1}^n\sum _{v=u+1}^n\sum _{d=1}^{\lfloor \frac{n}{v}\rfloor }[(u+v)|d][\gcd (u,v)=1]$$

由于在 $1\sim \lfloor \frac{n}{v}\rfloor$ 中，为 $u+v$ 的倍数的有 $\lfloor \frac{n}{v(u+v)}\rfloor$ 个，所以（上界我们先不予考虑）：

$$ans=\sum _{u=1}\sum _{v=u+1}\lfloor \frac{n}{v(u+v)}\rfloor [\gcd (u,v)=1]$$

然后可以反演一波：

$$\begin{array}{rl}ans& =\sum _{u=1}\sum _{v=u+1}\lfloor \frac{n}{v(u+v)}\rfloor \sum _{t|u,t|v}\mu (t)\\ & =\sum _{t=1}\mu (t)\sum _{u=1}\sum _{v=u+1}\lfloor \frac{n}{vt(ut+vt)}\rfloor \\ & =\sum _{t=1}\mu (t)\sum _{u=1}\sum _{v=u+1}\lfloor \frac{n}{v(u+v)t^2}\rfloor \end{array}$$

然后设 $k=u+v$，由于 $v>u$，所以 $v$ 的下界就是 $\lfloor \frac{k}{2}\rfloor +1$ 了，即：

$$ans=\sum _{t=1}\mu (t)\sum _{k=1}\sum _{v=\lfloor \frac{k}{2}\rfloor +1}\lfloor \frac{n}{vk{t}^2}\rfloor$$

讨论一下上界，$t\le \sqrt{n}$，$k\le \frac{\sqrt{2n}}{t}$，$v\le k-1$，这里可以比准确上界超出一点，反正超出部分贡献为 $0$。

用积分估计复杂度为 $O(n^{\frac{3}{4}})$。

---

code

#include<cmath>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define N 1000005
#define ll long long
using namespace std;
int n;
int sum,prime[N],mark[N],mu[N];
void linear_sieves(){
	mu[1]=1;
	for(int i=2;i<N;++i){
		if(!mark[i])  prime[++sum]=i,mu[i]=-1;
		for(int j=1;j<=sum&&i*prime[j]<N;++j){
			mark[i*prime[j]]=1;
			if(i%prime[j]==0)  {mu[i*prime[j]]=0;break;}
			mu[i*prime[j]]=-mu[i];
		}
	}
}
int main(){
	scanf("%d",&n);
	linear_sieves();
	ll ans=0;
	for(int d=1;(ll)d*d<=n;++d){
		if(!mu[d])  continue;
		int up=min((ll)n/d/d,(ll)sqrt(2ll*n)/d);
		for(int k=1;k<=up;++k){
			int T=n/d/d/k,u;
			for(int v=k/2+1;v<=min(k-1,T);v=u+1){
				u=min(k-1,T/(T/v));
				ans+=(ll)mu[d]*(u-v+1)*(T/v);
			}
		}
	}
	printf("%lld\n",ans);
	return 0;
}