【luogu P3327】约数个数和(莫比乌斯反演)

约数个数和

题目链接:luogu P3327

题目大意

给你 n,m 要你求 ∑i=1~n∑j=1~m d(ij),d 是约数个数。
多组数据。

思路

首先由一个比较神奇的东西:
\(d(ij)=\sum\limits_{x|i}\sum\limits_{y|j}[gcd(x,y)=1]\)

这里证明一下:
先看我们可以只看一个质数的部分。
假设它的因子的出现次数是 \(x\),\(i\) 中有 \(a\) 个,\(j\) 中有 \(b\) 个,那 \(ij\) 中就是 \(a+b\) 个。

如果 \(x\leqslant a\),那它在 \(i\) 中会出现被数到。
那如果 \(x>a\),我们可以在 \(j\) 中数一个 \(x-a\) 就表示满了之后再我那个这边的个数。

所以是对的。

然后就来化式子:
\(\sum\limits_{i=1}^n\sum\limits_{j=1}^md(ij)\)
\(=\sum\limits_{i=1}^n\sum\limits_{j=1}^m\sum\limits_{x|i}\sum\limits_{y|j}[gcd(x,y)=1]\)
\(=\sum\limits_{x=1}^n\sum\limits_{y=1}^m\sum\limits_{i=1}^{\frac{n}{x}}\sum\limits_{j=1}^{\frac{m} {y}}[gcd(x,y)=1]\)
\(=\sum\limits_{x=1}^n\sum\limits_{y=1}^m\left\lfloor\dfrac{n}{x}\right\rfloor\left\lfloor\dfrac{m} {y}\right\rfloor[gcd(x,y)=1]\)
\(=\sum\limits_{i=1}^n\sum\limits_{j=1}^m\left\lfloor\dfrac{n}{i}\right\rfloor\left\lfloor\dfrac{m} {j}\right\rfloor[gcd(i,j)=1]\)

然后莫反。
\(f_x=\sum\limits_{i=1}^n\sum\limits_{j=1}^m\left\lfloor\dfrac{n}{i}\right\rfloor\left\lfloor\dfrac{m} {j}\right\rfloor[gcd(i,j)=x]\)

\(g_i=\sum\limits_{x|i}f_i=\sum\limits_{i=1}^n\sum\limits_{j=1}^m\left\lfloor\dfrac{n}{i}\right\rfloor\left\lfloor\dfrac{m} {j}\right\rfloor[x|gcd(i,j)]\)
\(=\sum\limits_{i=1}^n\sum\limits_{j=1}^m\left\lfloor\dfrac{n}{i}\right\rfloor\left\lfloor\dfrac{m} {j}\right\rfloor\sum\limits_{x|i,x|j}1\)
\(=\sum\limits_{x}\sum\limits_{i=1}^{\left\lfloor\frac{n}{x}\right\rfloor}\sum\limits_{j=1}^{\left\lfloor\frac{m}{x}\right\rfloor}\left\lfloor\dfrac{n}{ix}\right\rfloor\left\lfloor\dfrac{m} {jx}\right\rfloor\)

\(f_i=\sum\limits_{i|x}\mu(\dfrac{x}{i})g_x\)
\(f_1=\sum\limits_{1|x}\mu(\dfrac{x}{1})g_x=\sum\limits_{i=1}^{\min(n,m)}\mu(i)g_i\)

那询问我们可以预处理 \(\mu\) 的前缀和然后用数论分块求 \(g_i\)。
(用数论分块预处理 \(\sum\limits_{i=1}^x\left\lfloor\dfrac{x}{i}\right\rfloor\))
(然后用数论分块搞 \(\sum\limits_{x}\) 的部分)

然后每次询问你就跑一次数论分块 \(O(\sqrt{n})\),所以询问的复杂度是 \(O(T\sqrt{n})\) 可以的。

代码

#include<cstdio>
#include<iostream>
#define ll long long

using namespace std;

int T, n, m, prime[50001];
ll f[50001], ans, miu[50001];
bool np[50001];

int main() {
	miu[1] = 1;
	for (int i = 2; i <= 50000; i++) {
		if (!np[i]) prime[++prime[0]] = i, miu[i] = -1;
		for (int j = 1; j <= prime[0] && i * prime[j] <= 50000; j++) {
			np[i * prime[j]] = 1;
			if (i % prime[j] == 0) {miu[i * prime[j]] = 0; break;}
				else miu[i * prime[j]] = -miu[i];
		}
	}
	for (int i = 1; i <= 50000; i++) miu[i] += miu[i - 1];
	
	for (int i = 1; i <= 50000; i++) {
		int r;
		for (int l = 1; l <= i; l = r + 1) {
			r = i / (i / l);
			f[i] += 1ll * (r - l + 1) * (i / l);
		}
	}
	
	scanf("%d", &T);
	while (T--) {
		scanf("%d %d", &n, &m);
		ans = 0;
		
		if (n > m) swap(n, m);
		
		int r;
		for (int l = 1; l <= n; l = r + 1) {
			r = min(n / (n / l), m / (m / l));
			ans += (miu[r] - miu[l - 1]) * f[n / l] * f[m / l];
		}
		
		printf("%lld\n", ans);
	}
	
	return 0;
}
上一篇:二项式反演学习笔记


下一篇:「 学习笔记 」二项式定理与组合恒等式