数论之 - 莫比乌斯反演

置顶 wangyiyang2

于 2020-05-05 17:06:51 发布

阅读量194

点赞数 1

分类专栏：读书笔记数论莫比乌斯反演

本文链接：https://blog.csdn.net/wangyiyang2/article/details/105934784

版权

读书笔记同时被 3 个专栏收录

8 篇文章 0 订阅

订阅专栏

数论

2 篇文章 0 订阅

订阅专栏

莫比乌斯反演

1 篇文章 0 订阅

订阅专栏

【学习笔记】数论之 - 莫比乌斯反演

$\mathrm{I.}$ 什么是莫比乌斯反演

$\mathrm{A.}$ 由 $F_{n}=\sum_{d|n}f_{d}$ → $f_n=\sum_{d|n} \mu(d)F(\frac{n}{d})$
$\mathrm{B.}$ 由 $F_{n}=\sum_{n|d}f_{d}$ → $f_n=\sum_{n|d} \mu(\frac{d}{n})F(d)$
$\mathrm{A,B}$ 为莫比乌斯反演的两种形式， $\mu$ 为莫比乌斯函数
【莫比乌斯函数】
- $\mu(1)=1$
- $\mu(n)=(-1)^k$ $(n=\prod_{1\leq i\leq k} P_i)$
- $\mu(n)=0$

$\mathrm{II.}$ 莫比乌斯的几道例题

例题1：洛谷P3455

求 $\sum_{i=1}^{n}\sum_{j=1}^{m}[\gcd(i,j)=k]$
有原式转换可得： $\sum_{i=1}^{\frac{n}{k}}\sum_{j=1}^{\frac{n}{k}}[\gcd(i,j)=1]$
使用莫比乌斯反演可得： $\sum_{i=1}^{\frac{n}{k}} \sum_{j=1}^{\frac{n}{k}}\sum_{d|\gcd(i,j)}\mu(d)$
为了方便（不需要去寻找 $\gcd(i,j)$ 的因数）我们去枚举 $d$ 得到 $\sum_{d=1}^{n}\mu(d)\sum_{i=1}^{\frac{n}{kd}}\sum_{j=1}^{\frac{n}{kd}}1$
我们就成功地消除了后面两个 $\sum$ ，得到最后的柿子： $\sum_{d=1}^{n}\mu(d) [\frac{n}{kd}][\frac{m}{kd}]$
然后我们先求出 $\mu(d)$ 的前缀和，对于后面一部分用高效的整除分块即可。

#include <bits/stdc++.h>
#define pb push_back
#define int long long
using namespace std;

inline int read()
{
	int sum=0,ff=1; char ch=getchar();
	while(!isdigit(ch))
	{
		if(ch=='-') ff=-1;
		ch=getchar();
	}
	while(isdigit(ch))
		sum=sum*10+(ch^48),ch=getchar();
	return sum*ff;
}

const int N=5e4+5;

int n,m,k,mu[N],sum[N],P[N],vis[N],cnt,ans;

inline void init()//莫比乌斯函数预处理
{
	mu[1]=1;
	for ( int i=2;i<=50000;i++ ) 
	{
		if(!vis[i]) 
		{
			P[++cnt]=i;
			mu[i]=-1;
		}
		for ( int j=1;j<=cnt,i*P[j]<=50000;j++ ) 
		{
			vis[i*P[j]]=1;
			if(!(i%P[j]))
			{
				mu[i*P[j]]=0;
				break;
			}
			mu[i*P[j]]=-mu[i];
		}
	}
	for ( int i=1;i<=50000;i++ ) sum[i]=sum[i-1]+mu[i];//前缀和
}

signed main()
{
	int Q=read();
	init();
	for (;Q--;)
	{
		n=read();
		m=read();
		k=read();
		ans=0;
		n/=k,m/=k;
		if(n<m) swap(n,m);//要保证m<n
		for ( int l=1,r;l<=m;l=r+1 ) 
		{
			r=min(n/(n/l),m/(m/l));//整除分块
			ans+=(sum[r]-sum[l-1])*(n/l)*(m/l);
		}
		printf("%lld\n",ans);
	}
	return 0;
}

例题2：洛谷P2522

求 $f_{i,j}=\sum_{i=a}^{b} \sum_{j=c}^{d} [\gcd(i,j)=k]$
其他部分与例题一求法类似，就是这代题目的下界更改了。我们要利用一些简单的容斥： $ans=f_{b,d}-f_{a-1,d}-f_{c-1,b}+f_{a-1,c-1}$
每个 $f_{i,j}$ 求法与例题一类似，不在赘述

#include <bits/stdc++.h>
#define pb push_back
#define int long long
using namespace std;

inline int read()
{
	int sum=0,ff=1; char ch=getchar();
	while(!isdigit(ch))
	{
		if(ch=='-') ff=-1;
		ch=getchar();
	}
	while(isdigit(ch))
		sum=sum*10+(ch^48),ch=getchar();
	return sum*ff;
}

const int N=5e4+5;

int mu[N],sum[N],vis[N],P[N],cnt,k;

inline void init()
{
	mu[1]=1;
	for ( int i=2;i<=50000;i++ ) 
	{
		if(!vis[i]) 
		{
			P[++cnt]=i;
			mu[i]=-1;
		}
		for ( int j=1;j<=cnt,i*P[j]<=50000;j++ ) 
		{
			vis[i*P[j]]=1;
			if(!(i%P[j])) 
			{
				mu[i*P[j]]=0;
				break;
			}
			mu[i*P[j]]=-1*mu[i];
		}
	}
	for ( int i=1;i<=50000;i++ ) sum[i]=sum[i-1]+mu[i];
}

inline int calc(int n,int m)
{
	n/=k,m/=k;
	if(n<m) swap(n,m);
	int ans=0; 
	for ( int l=1,r=0;l<=m;l=r+1 )
	{
		r=min(n/(n/l),m/(m/l));
		ans+=(sum[r]-sum[l-1])*(n/l)*(m/l);
	}
	return ans;
}

signed main()
{
	int Q=read();
	init();
	for (;Q--;)
	{
		int a,b,c,d;
		a=read();
		b=read();
		c=read();
		d=read();
		k=read();
		printf("%lld\n",calc(b,d)-calc(a-1,d)-calc(b,c-1)+calc(a-1,c-1));
	}
	return 0;
}

例题3：洛谷P2257

求 $\sum_{i=1}^{n}\sum_{j=1}^{m}[\gcd(i,j)∈质数]$
我们设 $P$ 为质数集
这道题目稍微复杂一点，我们先把原式变为 $\sum_{k∈P}\sum_{i=1}^{n}\sum_{j=1}^{m}[\gcd(i,j)=k]$
$\sum_{k∈P}\sum_{i=1}^{\frac{n}{k}}\sum_{j=1}^{\frac{n}{k}}[\gcd(i,j)=1]$
$\sum_{k∈P}\sum_{i=1}^{\frac{n}{k}}\sum_{j=1}^{\frac{n}{k}}\sum_{d|\gcd(i,j)}\mu(d)$
$\sum_{k∈P}\sum_{d=1}^{\frac{n}{k}}\mu(d)\sum_{i=1}^{\frac{n}{kd}}\sum_{j=1}^{\frac{n}{kd}}$
$\sum_{k∈P}\sum_{d=1}^{\frac{n}{k}}\mu(d)[\frac{n}{kd}][\frac{m}{kd}]$
我们再设 $T = k d$
则 $\sum_{k∈P}\sum_{d=1}^{\frac{n}{k}}\mu(d)[\frac{n}{T}][\frac{m}{T}]$
$\sum_{T=1}^{n}[\frac{n}{T}][\frac{m}{T}]\sum_{k∈P,k|T}\mu(\frac{T}{k})$
然后前面的东西可以用分块实现，后面的东西可以在莫比乌斯反演中前缀和预处理出来。

#include <bits/stdc++.h>
#pragma GCC optimize("Ofast",3,"inline")
#pragma GCC target("avx")
#define pb push_back
#define int long long
using namespace std;

inline int read()
{
	int sum=0,ff=1; char ch=getchar();
	while(!isdigit(ch))
	{
		if(ch=='-') ff=-1;
		ch=getchar();
	}
	while(isdigit(ch))
		sum=sum*10+(ch^48),ch=getchar();
	return sum*ff;
}

const int N=1e7+5;
const int MAXN=1e7;

int n,m,cnt,P[N],vis[N],mu[N],sum[N],f[N];

inline int min(int x,int y)
{
	if(x<y) return x;
	return y;
}

inline void init()
{
	mu[1]=1;
	for ( int i=2;i<=MAXN;i++ ) 
	{
		if(!vis[i])
		{
			P[++cnt]=i;
			mu[i]=-1;
		}
		for ( int j=1;j<=cnt,i*P[j]<=MAXN;j++ ) 
		{
			vis[i*P[j]]=1;
			if(!(i%P[j])) 
			{
				mu[i*P[j]]=0;
				break;
			}
			mu[i*P[j]]=-mu[i];
		}
	}
	for ( int i=1;i<=cnt;i++ ) 
		for ( int j=1;j*P[i]<=MAXN;j++ ) 
			f[j*P[i]]+=mu[j];
	for ( int i=1;i<=MAXN;i++ ) sum[i]=sum[i-1]+f[i];
}

signed main()
{
	int Q=read();
	init();
	for (;Q--;)
	{
		n=read();
		m=read();
		int Lim=min(n,m),ans=0;
		for ( int l=1,r;l<=Lim;l=r+1 ) 
		{
			int L=n/l,R=m/l;
			r=min(n/L,m/R);
			ans=(ans+(sum[r]-sum[l-1])*L*R);
		}
		printf("%lld\n",ans);
	}
	return 0;
}

例题4：洛谷P3911

求 $\sum_{i=1}^{n}\sum_{j=1}^{n} lcm(a_i,a_j)$
考虑到 $a_i\leq 5e4$ ，我们转变原柿变为： $\sum_{i=1}^{mx}\sum_{j=1}^{mx}\frac{ij\times c_ic_j}{\gcd(i,j)}$
上式中的 $mx=\max a_i$ ， $c_i$ 为 $i$ 出现过的次数
按照套路变形： $\sum_{d=1}^{mx}\sum_{i=1}^{mx}\sum_{j=1}^{mx}[\gcd(i,j)=d]c_ic_j\frac{ij}{d}$
$\sum_{d=1}^{mx}\sum_{i=1}^{\frac{mx}{d}}\sum_{j=1}^{\frac{mx}{d}}[\gcd(i,j)=1]c_{id}c_{jd}\times ijd$
$\sum_{d=1}^{mx}\sum_{i=1}^{\frac{mx}{d}}\sum_{j=1}^{\frac{mx}{d}}\sum_{k|\gcd(i,j)}\mu(k)c_{id}c_{jd}\times ijd$
$\sum_{d=1}^{mx}\sum_{k=1}^{\frac{mx}{d}}\mu(k)k^2\sum_{i=1}^{\frac{mx}{kd}}\sum_{j=1}^{\frac{mx}{kd}}c_{ikd}c_{jkd}\times ijd$
$\sum_{d=1}^{mx}\sum_{kd=1}^{mx}\mu(k)k^2\sum_{i=1}^{\frac{mx}{kd}}\sum_{j=1}^{\frac{mx}{kd}}c_{ikd}c_{jkd}\times ijd$
设 $T = k d$ ，则
$\sum_{d=1}^{mx}\sum_{T=1}^{mx}\mu(k)k^2\sum_{i=1}^{\frac{mx}{T}}\sum_{j=1}^{\frac{mx}{T}}c_{iT}c_{jT}\times ijd$
考虑枚举 $T$
那么 $\sum_{T=1}^{mx}\sum_{k|T}(k\times \mu(k))\sum_{i=1}^{\frac{mx}{T}}(c_{iT}\times i)$
现在我们对于前半部分预处理一下，后面部分暴力算一下就可以了。

#include <bits/stdc++.h>
#pragma GCC optimize("Ofast",3,"inline")
#pragma GCC target("avx")
#define pb push_back
using namespace std;

inline int read()
{
	int sum=0,ff=1; char ch=getchar();
	while(!isdigit(ch))
	{
		if(ch=='-') ff=-1;
		ch=getchar();
	}
	while(isdigit(ch))
		sum=sum*10+(ch^48),ch=getchar();
	return sum*ff;
}

const int N=5e4+5;

int n,m,cnt,P[N],vis[N],mu[N],a[N]; 
long long f[N];

inline int min(int x,int y)
{
	if(x<y) return x;
	return y;
}

inline void init(int MAXN)
{
	mu[1]=1;
	for ( int i=2;i<=MAXN;i++ ) 
	{
		if(!vis[i])
		{
			P[++cnt]=i;
			mu[i]=-1;
		}
		for ( int j=1;j<=cnt,i*P[j]<=MAXN;j++ ) 
		{
			vis[i*P[j]]=1;
			if(!(i%P[j])) 
			{
				mu[i*P[j]]=0;
				break;
			}
			mu[i*P[j]]=-mu[i];
		}
	}
	for ( int i=1;i<=MAXN;i++ ) 
		for ( int j=i;j<=MAXN;j+=i ) 
			f[j]+=i*mu[i];
}

int main()
{
	int mx=0;
	n=read();
	for ( int i=1;i<=n;i++ ) 
	{
		int x=read();
		a[x]++;
		mx=max(mx,x);
	}
	init(mx); 
	long long ans=0;
	for ( int i=1;i<=mx;i++ ) 
	{
		long long now=0;
		for ( int j=1;j<=mx/i;j++ ) 
			now+=(a[i*j]*j);
		ans+=i*f[i]*now*now;
	}
	printf("%lld\n",ans);
	return 0;
}
/*
2
2 3
17
*/

wangyiyang2

关注

1
点赞
踩
0

收藏

觉得还不错? 一键收藏
0
评论
数论之 - 莫比乌斯反演

【学习笔记】数论之 - 莫比乌斯反演I.\mathrm{I.}I.什么是莫比乌斯反演A.\mathrm{A.}A. 由Fn=∑d∣nfdF_{n}=\sum_{d|n}f_{d}Fn=∑d∣nfd → fn=∑d∣nμ(d)F(nd)f_n=\sum_{d|n} \mu(d)F(\frac{n}{d})fn=∑d∣nμ(d)F(dn)B.\mathrm{B.}B. 由Fn...
复制链接

扫一扫