LuoguP3327 [SDOI2015]约数个数和

题目描述

设 $d(x)$ 为 $x$ 的约数个数，给定 $n,m$ 求

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

引理：

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

证明：

对于 $ij$ 的某一个约数 $s=p_1^{c_1}{p}_2^{c_2}...p_k^{c_k}$
对于 $s$ 的某一个素因子 $p$ 设其次数为 $c$，在 $i$ 中次数为 $a$ ，在 $j$ 中次数为 $b$ 。

• $c\le a$，在 $(x,y)=(c,0)$ 时，形成唯一映射
• $c>a$，在$(x,y)=(0,c-a)$ 时，形成唯一映射
  故有且仅有一组 $(x,y)$ 与素因子 $p$ 的次数对应
  注意到 $x,y$ 中至少有一个 $0$，所以用 $gcd$ 来限制即可

解法

$$\sum _{i=1}^n\sum _{j=1}^m\sum _{x|i}\sum _{y|j}[gcd(x,y)=1]$$
$$\sum _{i=1}^n\sum _{j=1}^m\sum _{x|i}\sum _{y|j}\sum _{d|x,d|y}\mu (d)$$
$$\sum _{x=1}^n\sum _{y=1}^m\lfloor \frac{n}{x}\rfloor \lfloor \frac{m}{y}\rfloor \sum _{d|x,d|y}\mu (d)$$
$$\sum _{d=1}^{min(n,m)}\mu (d)\sum _{x=1}^{\lfloor \frac{n}{d}\rfloor }\sum _{y=1}^{\lfloor \frac{m}{d}\rfloor }\lfloor \frac{n}{dx}\rfloor \lfloor \frac{m}{dy}\rfloor$$
$$\sum _{d=1}^{min(n,m)}\mu (d)(\sum _{x=1}^{\lfloor \frac{n}{d}\rfloor }\lfloor \frac{n}{dx}\rfloor )(\sum _{y=1}^{\lfloor \frac{m}{d}\rfloor }\lfloor \frac{m}{dy}\rfloor )$$
令 $f(x)=\sum _{i=1}^x\lfloor \frac{x}{i}\rfloor$
$$\sum _{d=1}^{min(n,m)}\mu (d)(\sum _{x=1}^{\lfloor \frac{n}{d}\rfloor }\lfloor \frac{\lfloor \frac{n}{d}\rfloor }{x}\rfloor )(\sum _{y=1}^{\lfloor \frac{m}{d}\rfloor }\lfloor \frac{\lfloor \frac{m}{d}\rfloor }{y}\rfloor )$$

$$\sum _{d=1}^{min(n,m)}\mu (d)f(\lfloor \frac{n}{d}\rfloor )f(\lfloor \frac{m}{d}\rfloor )$$
到这里就可以直接对 $d$ 数论分块。

算 $f(x)$可以用差分的思想，考虑 $i$ 的贡献。

$i$ 对 $f(i$ ~ $2i-1)$ 有 $1$ 的贡献
对 $f(2i$ ~ $3i-1)$ 有 $2$ 的贡献
$...$
差分一下，给 $f(ki)+=1$ ，对于 $i$ 共进行了 $\frac{n}{i}$ 次操作

复杂度 $\mathcal{O}(nln)$ 预处理即可。

代码：

#include<cstdio>
#include<map>
#include<algorithm>
using namespace std;
typedef long long LL;
const int P=50005;

int prime[P],pcnt;
bool vis[P];
int mu[P],sum[P],f[P];
void Init() {
	sum[1]=mu[1]=1;
	for(int i=2;i<P;i++) {
		if(!vis[i]) mu[prime[++pcnt]=i]=-1;
		for(int j=1;j<=pcnt&&i*prime[j]<P;j++) {
			vis[i*prime[j]]=true;
			if(i%prime[j]==0) break;
			mu[i*prime[j]]=-mu[i];
		}
		sum[i]=sum[i-1]+mu[i];
	}
	for(int i=1;i<P;i++)
		for(int j=i;j<P;j+=i)
			f[j]++;
	for(int i=1;i<P;i++)
		f[i]+=f[i-1];
}
inline LL Solve(int A,int B) {
	LL ret=0;
	for(int l=1,r;l<=A;l=r+1) {
		r=min(A/(A/l),B/(B/l));
		ret=ret+1ll*(sum[r]-sum[l-1])*f[A/l]*f[B/l];
	}
	return ret;
}
int T,n,m;
int main() {
	Init();
	scanf("%d",&T);
	for(int Cas=1;Cas<=T;Cas++) {
		scanf("%d %d",&n,&m);
		if(n>m) swap(n,m);
		printf("%lld\n",Solve(n,m));
	}
	return 0;
}