P3327 [SDOI2015]约数个数和(莫比乌斯反演)

题目描述:

设\(d(x)\)为\(x\)的约数个数,给定\(n,m\),求\(\sum_{i=1}^n\sum_{j=1}^md(ij)\)

首先有一个前置知识:

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

所以原式等于:

\(\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|gcd(x,y)}\mu(d)\)

交换求和顺序:

\(\sum_{d=1}^n\mu(d)\sum_{x=1}^n\sum_{y=1}^m[d|gcd(x,y)][n/x][m/y]\)

然后可以把枚举\(x,y\)改为枚举\(d\)的倍数,把\([d|gcd(x,y)]\)这个玩意去掉:

\(\sum_{d=1}^n\mu(d)\sum_{x=1}^{n/d}\sum_{y=1}^{m/d}[n/dx][m/dy]\)

把与x有关的项,与y有关的项合并,有

\(\sum_{d=1}^n\mu(d)\sum_{x=1}^{n/d}[n/dx]\sum_{y=1}^{m/d}[m/dy]\)

然后套莫反笔记的引理1:

\(\sum_{d=1}^n\mu(d)\sum_{x=1}^{n/d}[[n/d]/x]\sum_{y=1}^{m/d}[[m/d]/y]\)

这里我们设\(p=n/d,q=m/d\):

\(\sum_{d=1}^n\mu(d)\sum_{x=1}^p[p/x]\sum_{y=1}^q[q/y]\)

设\(f(x)=\sum_{i=1}^x[x/i]\)

\(\sum_{d=1}^n\mu(d)f(p)f(q)\)

这里\(p=[n/d]\),\(q=[m/d]\),所以可以对\(d\)进行数论分块,同一个块内的\(p\)和\(q\)可以快速计算。

现在预处理求\(f(x)\)。暴力时间复杂度是\(O(n^2)\)的。

观察\(f(x)=\sum_{i=1}^x[x/i]\),同一块内\(x/i\)的值不变,所以对每个\(x\)可以数论分块求。

时间复杂度\(O(n\sqrt{n}+T\sqrt{n})\)。

#include<bits/stdc++.h>
using namespace std;
const int maxn=1e5+100;
typedef long long ll;
int vis[maxn],pri[maxn],mu[maxn],sum[maxn],cnt;
ll f[maxn];
void getMu (int n) {
	mu[1]=1;
	for (int i=2;i<=n;i++) {
		if (!vis[i]) {
			mu[i]=-1;
			pri[++cnt]=i;
		}
		for (int j=1;j<=cnt&&i*pri[j]<=n;j++) {
			vis[i*pri[j]]=1;
			if (i%pri[j]==0) break;
			else mu[i*pri[j]]=-mu[i];
		}
	}
	for (int i=1;i<=n;i++) sum[i]=sum[i-1]+mu[i];
	for (int i=1;i<=n;i++) {
		for (int l=1,r;l<=i;l=r+1) {
			r=i/(i/l);
			f[i]+=1ll*(r-l+1)*(i/l);
		}
	}
}
int main () {
	int _;
	getMu(5e4);
	scanf("%d",&_);
	while (_--) {
		int n,m;
		scanf("%d%d",&n,&m);
		if (n>m) swap(n,m);
		ll ans=0;
		for (int l=1,r;l<=n;l=r+1) {
			r=min(n/(n/l),m/(m/l));
			ans+=1ll*f[n/l]*f[m/l]*(sum[r]-sum[l-1]);
		}
		printf("%lld\n",ans);
	}
}

上一篇:强化学习:浅析DDPG


下一篇:TrickGCD HDU-6053 莫比乌斯反演 容斥