莫比乌斯反演习题

xishanmeigao

于 2023-04-03 15:42:51 发布

阅读量459

点赞数

分类专栏：学习总结题解文章标签： c++ 算法莫比乌斯反演

本文链接：https://blog.csdn.net/xishanmeigao2918/article/details/129887623

版权

题解同时被 2 个专栏收录

47 篇文章 10 订阅

订阅专栏

学习总结

11 篇文章 2 订阅

订阅专栏

$T i p s$ ：本文默认 $n\leq m,\: a\leq b$

[POI2007] ZAP-Queries

（题目传送门）

题意

给出 $a, b, d$ ，求满足 $\leq x \leq a$ ， $\leq y \leq b$ ，且 $\gcd(x,y)=d$ 的二元组 $(x, y)$ 的数量。

$\leq n \leq 5 \times 10^4$ ， $\leq d \leq a,b \leq 5 \times 10^4$ 。

解题思路

原题要求的是
$\sum\limits_{i=1}^a\sum\limits_{j=1}^b{[\gcd(i,j)=d]}$
将 $i, j$ 除以 $d$ 得
$=\sum\limits_{i=1}^{\left\lfloor\frac{a}{d}\right\rfloor}\sum\limits_{j=1}^{\left\lfloor\frac{b}{d}\right\rfloor}{[\gcd(i,j)=1]}$
由莫比乌斯反演得
$=\sum\limits_{i=1}^{\left\lfloor\frac{a}{d}\right\rfloor}\sum\limits_{j=1}^{\left\lfloor\frac{b}{d}\right\rfloor}\sum\limits_{k|\gcd(i,j)}{\mu(k)}$
再将 $i, j$ 除以 $k$ ，这样就不关心 $k$ 是否为 $i, j$ 的约数，可以直接枚举 $k$ ：

$=\sum\limits_{k=1}^{\left\lfloor\frac{a}{d}\right\rfloor}\mu(k){\left\lfloor\frac{a}{kd}\right\rfloor}{\left\lfloor\frac{b}{kd}\right\rfloor}$
这个式子的后半部分可以整除分块来优化，前半部分预处理前缀和即可。在代码实现时，可以预先让 $a, b$ 同除以 $d$ ，方便实现

代码

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

const int N=50010;

int T,a,b,d,n,m;
int prime[N],v[N],tot,mu[N],sum[N];
LL ans;

void primes()
{
	mu[1]=1;
	for(int i=2; i<=50000; i++)
	{
		if(!v[i])
		{
			v[i]=i;
			prime[++tot]=i;
			mu[i]=-1;
		}
			
		for(int j=1; j<=tot; j++)
		{
			if(prime[j]>v[i] || prime[j]>50000/i)
				break;
			v[i*prime[j]]=prime[j];
			if(i%prime[j]==0)
				break;
			mu[prime[j]*i]=-mu[i];
		}
	}
	
	for(int i=1; i<=50000; i++)
		sum[i]=sum[i-1]+mu[i];
}
 	
int main()
{
	primes();
	
    scanf("%d",&T);
    while(T--)
    {
    	scanf("%d%d%d",&a,&b,&d);
    	n=a/d;  m=b/d;
    	if(n>m)
    		swap(n,m);
    	
    	ans=0;
    	for(int l=1,r; l<=n; l=r+1)
    	{
    		r=min(n/(n/l),m/(m/l));
    		ans+=1LL*(sum[r]-sum[l-1])*1LL*(n/l)*1LL*(m/l);
		}
		
		printf("%lld\n",ans);
	}
    
    return 0;
}

[HAOI2011] Problem b

（题目传送门）

题意

对于给出的 $n$ 个询问，每次求有多少个数对 $(x, y)$ ，满足 $\le x \le b$ ， $\le y \le d$ ，且 $\gcd(x,y) = k$ ， $\gcd(x,y)$ 函数为 $x$ 和 $y$ 的最大公约数。

$\le n,k \le 5 \times 10^4$ ， $\le a \le b \le 5 \times 10^4$ ， $\le c \le d \le 5 \times 10^4$ 。

解题思路

类比上面一题 ZAP，设 $f (n, m)$ 表示满足 $\leq x \leq n$ ， $\leq y \leq m$ ，且 $\gcd(x,y)=k$ 的二元组 $(x, y)$ 的数量
类似二维前缀和，容易得出
$an s = f (b, d) - f (a - 1, d) - f (b, c - 1) + f (a - 1, c - 1)$
同样，可以一开始先令 $a, b, c, d$ 除以 $k$ ，方便代码实现

代码

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

const int N=50010;

int T,a,b,c,d,k;
int prime[N],v[N],tot,mu[N],sum[N];
LL ans;

void primes()
{
	mu[1]=1;
	for(int i=2; i<=50000; i++)
	{
		if(!v[i])
		{
			v[i]=i;
			prime[++tot]=i;
			mu[i]=-1;
		}
			
		for(int j=1; j<=tot; j++)
		{
			if(prime[j]>v[i] || prime[j]>50000/i)
				break;
			v[i*prime[j]]=prime[j];
			if(i%prime[j]==0)
				break;
			mu[prime[j]*i]=-mu[i];
		}
	}
	
	for(int i=1; i<=50000; i++)
		sum[i]=sum[i-1]+mu[i];
}

LL f(int n,int m)
{
	if(n>m)
		swap(n,m);
	LL res=0;
	for(int l=1,r; l<=n; l=r+1)
   	{
    	r=min(n/(n/l),m/(m/l));
    	res+=1LL*(sum[r]-sum[l-1])*1LL*(n/l)*1LL*(m/l);
	}
	
	return res;
}
 	
int main()
{
	primes();
	
    scanf("%d",&T);
    while(T--)
    {
    	scanf("%d%d%d%d%d",&a,&b,&c,&d,&k);
    	
    	ans=0;
    	ans=f(b/k,d/k)-f((a-1)/k,d/k)-f(b/k,(c-1)/k)+f((a-1)/k,(c-1)/k);
		
		printf("%lld\n",ans);
	}
    
    return 0;
}

YY的GCD

（题目传送门）

题意

$T$ 组数据，每组数据给定 $N, M$ ，求 $\leq x \leq N$ ， $\leq y \leq M$ 且 $\gcd(x, y)$ 为质数的 $(x, y)$ 有多少对。

$T = 10^4$ ， $\leq 10^7$ 。

解题思路

设 $p r im es$ 表示质数集合
原题要求的是
$ans=\sum\limits_{k\in primes}\sum\limits_{i=1}^{n}\sum\limits_{j=1}^m{[\gcd(i,j)=k]}$
同除以 $k$ 得：
$=\sum\limits_{k\in primes}\sum\limits_{i=1}^{\left\lfloor\frac{n}{k}\right\rfloor}\sum\limits_{j=1}^{\left\lfloor\frac{m}{k}\right\rfloor}{[\gcd(i,j)=k]}$
由莫比乌斯反演得
$=\sum\limits_{k\in primes}\sum\limits_{i=1}^{\left\lfloor\frac{n}{k}\right\rfloor}\sum\limits_{j=1}^{\left\lfloor\frac{m}{k}\right\rfloor}\sum\limits_{d|\gcd(i,j)}{\mu(d)}$
我们枚举 $d$ 得：
$=\sum\limits_{k\in primes}\sum\limits_{d=1}^{\left\lfloor\frac{n}{k}\right\rfloor}{\mu(d)}\left\lfloor\frac{n}{kd}\right\rfloor\left\lfloor\frac{m}{kd}\right\rfloor$
设 $T = k d$ ，把 $T$ 提到前面去
$=\sum\limits_{T=1}^n\left\lfloor\frac{n}{T}\right\rfloor\left\lfloor\frac{m}{T}\right\rfloor\sum\limits_{k|T,k\in primes}{\mu(\frac{T}{k})}$
对于后半部分式子，我们可以在线性筛后预处理出来，再做一次整除分块即可求出答案

代码

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

const int N=10000010;

int T,n,m;
int prime[N],v[N],tot,mu[N];
LL sum[N],g[N];

void primes()
{
	mu[1]=1;
	for(int i=2; i<=1e7; i++)
	{
		if(!v[i])
		{
			v[i]=i;
			prime[++tot]=i;
			mu[i]=-1;
		}
			
		for(int j=1; j<=tot; j++)
		{
			if(prime[j]>v[i] || prime[j]>1e7/i)
				break;
			v[i*prime[j]]=prime[j];
			if(i%prime[j]==0)
				break;
			mu[prime[j]*i]=-mu[i];
		}
	}
	
	for(int i=1; i<=tot; i++)
	{
		for(int j=1; j*prime[i]<=1e7; j++)
			g[prime[i]*j]+=mu[j]; 
	}
	
	for(int i=1; i<=1e7; i++)
		sum[i]=sum[i-1]+g[i];
}

LL cacl(int n,int m)
{
	if(n>m)
		swap(n,m);
	
	LL res=0; 
	for(int l=1,r; l<=n; l=r+1)
	{
		r=min(n/(n/l),m/(m/l));
		res+=1LL*(sum[r]-sum[l-1])*(n/l)*(m/l);
	}
	
	return res;
}
 	
int main()
{
	primes();
	
    scanf("%d",&T);
    while(T--)
    {
    	scanf("%d%d",&n,&m);
		printf("%lld\n",cacl(n,m));
	}
    
    return 0;
}

于神之怒加强版

（题目传送门）

题意

给定 $n, m, k$ ，计算

$\sum_{i=1}^n \sum_{j=1}^m \gcd(i,j)^k$

对 $10^9 + 7$ 取模的结果。

对于全部的测试点，保证 $\leq T \leq 2 \times 10^3$ ， $\leq n, m, k \leq 5 \times 10^6$ 。

解题思路

直接开始推式子
$ans=\sum_{d=1}^n\sum_{i=1}^n\sum_{j=1}^m{d^k[\gcd(i,j)=d]}$
除以 $d$ 得：
$=\sum_{d=1}^n\sum_{i=1}^{\left\lfloor\frac{n}{d}\right\rfloor}\sum_{j=1}^{\left\lfloor\frac{m}{d}\right\rfloor}{d^k[\gcd(i,j)=1]}$
$=\sum_{d=1}^n{d^k}\sum_{i=1}^{\left\lfloor\frac{n}{d}\right\rfloor}\sum_{j=1}^{\left\lfloor\frac{m}{d}\right\rfloor}\sum_{x|\gcd(i,j)}{\mu(x)}$
枚举 $x$ 得：
$\sum_{d=1}^n{d^k}\sum_{x=1}^{\left\lfloor\frac{n}{d}\right\rfloor}{\mu(x)\left\lfloor\frac{n}{dx}\right\rfloor\left\lfloor\frac{m}{dx}\right\rfloor}$
类似上一题的优化，设 $T = d x$ 得：
$\sum_{T=1}^n{\left\lfloor\frac{n}{T}\right\rfloor\left\lfloor\frac{m}{T}\right\rfloor}\sum_{d|T}{d^k\mu(\frac{T}{d})}$
观察到， $\sum\limits_{d|T}{d^k\mu(\frac{T}{d})}$ 是狄利克雷卷积的形式，而 $f(d)=d^k$ 和 $\mu(\frac{T}{d})$ 又是 积性函数，所以 $g(T)=\sum\limits_{d|T}{d^k\mu(\frac{T}{d})}$ 也是积性函数。因此，我们可以考虑线性筛预处理出 $g$ 的值
设 $T$ 质因数分解为 $T=\prod\limits_{i=1}^q{p_i^{c_i}}$ ，则
$g(T)=\prod\limits_{i=1}^q{g(p_i^{c_i})}$
$=\prod\limits_{i=1}^q\sum_{d|p_i^{c_i}}{d^k\mu(\frac{p_i^{c_i}}{d})}$
观察到，当 $d=1,p_i,pi^2\dots p_i^{c_i-2}$ 时， $\mu(\frac{p_i^{c_i}}{d})=0$ ，所以
$g(T)=\prod\limits_{i=1}^q{p_i^{(c_i-1)k}\times\mu(p_i)}+p_i^{c_ik}\times \mu(1)$
$=\prod\limits_{i=1}^q{p_i^{(c_i-1)k}\times(p_i^k-1)}$
这个式子是可以预处理出来的
所以
$ans=\sum_{T=1}^n{\left\lfloor\frac{n}{T}\right\rfloor\left\lfloor\frac{m}{T}\right\rfloor}g(T)$
利用整除分块即可求解

代码

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

const int N=5000010;
const int MOD=1e9+7;

int T,k,n,m;
int prime[N],v[N],tot;
LL g[N],mi[N],sum[N];

LL ksm(int a,int b)
{
	if(b==0)
		return 1;
	if(b==1)
		return a;
		
	LL tmp=ksm(a,b/2);
	if(b%2==0)
		return (tmp*tmp)%MOD;
	else
		return ((tmp*tmp)%MOD*(LL)a)%MOD;
}

void primes()
{
	g[1]=1;
	for(int i=2; i<=5e6; i++)
	{
		if(!v[i])
		{
			v[i]=i;
			prime[++tot]=i;
			mi[i]=ksm(i,k);
			g[i]=(mi[i]-1+MOD)%MOD; //i为质数时g(i)=i^k-1
		}
			
		for(int j=1; j<=tot; j++)
		{
			if(prime[j]>v[i] || prime[j]>5e6/i)
				break;
			v[i*prime[j]]=prime[j];
			
			if(i%prime[j]==0)
				g[i*prime[j]]=(g[i]*mi[prime[j]])%MOD; //当i和prime[j]不互质时，读者可自行证明该结论正确性
			else
				g[i*prime[j]]=(g[i]*g[prime[j]])%MOD;
		}
	}
	
	for(int i=1; i<=5e6; i++)
		sum[i]=sum[i-1]+g[i];
}

LL cacl(int n,int m)
{
	if(n>m)
		swap(n,m);
	
	LL res=0; 
	for(int l=1,r; l<=n; l=r+1)
	{
		r=min(n/(n/l),m/(m/l));
		res=(res+((sum[r]-sum[l-1]+MOD)%MOD)*(n/l)%MOD*(m/l)%MOD)%MOD;
	}
	
	return res;
}
 	
int main()
{
	scanf("%d%d",&T,&k);
	
	primes();
	
    while(T--)
    {
    	scanf("%d%d",&n,&m);
		printf("%lld\n",cacl(n,m));
	}
    
    return 0;
}

[国家集训队] Crash的数字表格 / JZPTAB

（题目传送门）

题意

给定 $n, m$ ，计算
$\sum_{i=1}^n\sum_{j=1}^m{\operatorname{lcm}(i,j)}$
$1\leq n,m \leq 10^7$

解题思路

对 $\operatorname{lcm}(i,j)$ 变换一下得：
$ans=\sum_{i=1}^n\sum_{j=1}^m{\frac{ij}{\gcd(i,j)}}$
枚举 $\gcd(i,j)$ 得：
$=\sum_{i=1}^n\sum_{j=1}^m\sum_{d=1}^n{\frac{ij}{d}[\gcd(i,j)=d]}$
除以 $d$ 得：
$=\sum_{d=1}^n\sum_{i=1}^{\left\lfloor\frac{n}{d}\right\rfloor}\sum_{j=1}^{\left\lfloor\frac{m}{d}\right\rfloor}{ij\times d[\gcd(i,j)=1]}$
$=\sum_{d=1}^n{d}\sum_{i=1}^{\left\lfloor\frac{n}{d}\right\rfloor}\sum_{j=1}^{\left\lfloor\frac{m}{d}\right\rfloor}{ij[\gcd(i,j)=1]}$
把后半部分式子单独提出来，设：
$f(a,b)=\sum\limits_{i=1}^{a}\sum\limits_{j=1}^{b}{ij[\gcd(i,j)=1]}$
化简得：
$f(a,b)=\sum\limits_{i=1}^{a}\sum\limits_{j=1}^{b}{ij\sum_{k|\gcd(i,j)}{\mu(k)}}$
$=\sum_{k=1}^{a}{\mu(k)}\sum_{i=1}^{\left\lfloor\frac{a}{k}\right\rfloor}\sum_{j=1}^{\left\lfloor\frac{b}{k}\right\rfloor}{ij\times k^2}$
$=\sum_{k=1}^{a}{\mu(k)\times k^2}\sum_{i=1}^{\left\lfloor\frac{a}{k}\right\rfloor}\sum_{j=1}^{\left\lfloor\frac{b}{k}\right\rfloor}{ij}$
设
$g(a,b)=\sum_{i=1}^{a}\sum_{j=1}^{b}{ij}$
化简得：
$=\frac{(1+a)a}{2}\times \frac{(b+1)b}{2}$
所以，求出 $g (a, b)$ 的时间复杂度为 $O (1)$
将 $g (a, b)$ 代回 $f (a, b)$ 中得：
$f(a,b)=\sum_{k=1}^{a}{\mu(k)\times k^2\times g(\left\lfloor\frac{a}{k}\right\rfloor,\left\lfloor\frac{b}{k}\right\rfloor)}$
我们利用整除分块即可求出 $f (a, b)$
将 $f (a, b)$ 代回 $an s$ 得：
$ans=\sum_{d=1}^n{d}\times f(\left\lfloor\frac{n}{d}\right\rfloor,\left\lfloor\frac{m}{d}\right\rfloor)$
再做一次整除分块即可求解

代码

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

const int N=10000010;
const int MOD=20101009;

int n,m;
int prime[N],v[N],tot,mu[N];
LL sum[N];

void primes()
{
	mu[1]=1;
	for(int i=2; i<=1e7; i++)
	{
		if(!v[i])
		{
			v[i]=i;
			prime[++tot]=i;
			mu[i]=-1;
		}
			
		for(int j=1; j<=tot; j++)
		{
			if(prime[j]>v[i] || prime[j]>1e7/i)
				break;
			v[i*prime[j]]=prime[j];
			if(i%prime[j]==0)
				break;
			mu[i*prime[j]]=-mu[i];
		}
	}
	
	for(int i=1; i<=1e7; i++)
		sum[i]=(sum[i-1]+1LL*(mu[i]+MOD)%MOD*i%MOD*i%MOD)%MOD;
}

LL g(int a,int b)
{
	return (1LL*a*(a+1)/2%MOD)*(1LL*b*(b+1)/2%MOD)%MOD;
}

LL f(int a,int b)
{
	if(a>b)
		swap(a,b);
		
	LL res=0;
	for(int l=1,r; l<=a; l=r+1)
	{
		r=min(a/(a/l),b/(b/l));
		res=(res+(sum[r]-sum[l-1]+MOD)%MOD*g(a/l,b/l)%MOD)%MOD;
	}
	
	return res;
}

LL calc(int a,int b)
{
	if(a>b)
		swap(a,b);
		
	LL res=0;
	for(int l=1,r; l<=a; l=r+1)
	{
		r=min(a/(a/l),b/(b/l));
		res=(res+1LL*(r-l+1)*(l+r)/2%MOD*f(a/l,b/l)%MOD)%MOD;
	}
	
	return res;
}
 		
int main()
{
	primes();
	
	scanf("%d%d",&n,&m);
	printf("%lld",calc(n,m));
    
    return 0;
}

[SDOI2014] 数表

（题目传送门）

题意

有一张 $n\times m$ 的数表，其第 $i$ 行第 $j$ 列（ $1\le i\le n$ ， $1\le j\le m$ ）的数值为能同时整除 $i$ 和 $j$ 的所有自然数之和。给定 $a$ ，计算数表中不大于 $a$ 的数之和。

$1\le n,m\le 10^5$ ， $1\le Q\le 2\times 10^4$ 。

解题思路

设 $f (x)$ 表示 $x$ 的因数之和
则
$ans=\sum_{i=1}^n\sum_{j=1}^m{f(\gcd(i,j))\times[f(\gcd(i,j))\leq a]}$
枚举 $\gcd(i,j)$ 得：
$=\sum_{d=1}^n\sum_{i=1}^n\sum_{j=1}^m{f(d)\times [\gcd(i,j)=d]\times[f(d)\leq a]}$
$=\sum_{d=1}^nf(d)\times[f(d)\leq a]\sum_{i=1}^{\left\lfloor\frac{n}{d}\right\rfloor}\sum_{j=1}^{\left\lfloor\frac{m}{d}\right\rfloor}{[\gcd(i,j)=1]}$
由莫比乌斯反演得：
$=\sum_{d=1}^nf(d)\times[f(d)\leq a]\sum_{i=1}^{\left\lfloor\frac{n}{d}\right\rfloor}\sum_{j=1}^{\left\lfloor\frac{m}{d}\right\rfloor}\sum_{k|\gcd(i,j)}{\mu(k)}$
$=\sum_{d=1}^nf(d)\times[f(d)\leq a]\sum_{k=1}^{\left\lfloor\frac{n}{d}\right\rfloor}{\mu(k)}\left\lfloor\frac{n}{kd}\right\rfloor\left\lfloor\frac{m}{kd}\right\rfloor$
套路设 $T = k d$ 得：
$=\sum_{T=1}^n{\left\lfloor\frac{n}{T}\right\rfloor\left\lfloor\frac{m}{T}\right\rfloor}\sum_{x|T}{\mu(\frac{T}{x})\times f(x)\times [f(x)\leq a]}$
$f (x)$ 可以在 $O(n\ln n)$ 的时间预处理出来
因为有 $a$ 的限制，当 $a$ 变大时，后半部分式子也在改变。所以考虑将询问离线，按 $a$ 的值从小到大排序，每次 $a$ 变大时都类似插入操作，考虑用树状数组维护
设 $g(T)=\sum\limits_{x|T}{\mu(\frac{T}{x})\times f(x)}$ ，对于所有 $f(x)\leq a$ 来说，它会对 $g(x),g(2x)\dots$ 产生贡献，所以给每个 $g$ 都加上 $\mu(\frac{T}{x})\times f(x)$ 即可

代码

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

const int N=100010,Q=20010;
const LL MOD=2147483648;

struct node
{
	int dat;
	int id;
}f[N];
struct que
{
	int n,m,s,id;
}a[Q];

int T;
int prime[N],v[N],tot,mu[N];
LL c[N],ans[N];

bool cmp(que xx,que yy)
{
	return xx.s<yy.s;
}

bool cmp2(node xx,node yy)
{
	return xx.dat<yy.dat;
}

void add(int x,LL y)
{
	for(; x<=1e5; x+=(x&-x))
		c[x]=(c[x]+y)%MOD;
}

LL ask(int x)
{
	LL s=0;
	for(; x; x-=(x&-x))
		s=(c[x]+s)%MOD;
	return s;
}

void update(int x,LL y)
{
	for(int i=1; i*x<=1e5; i++)
		add(i*x,1LL*mu[i]*y);
}

void primes()
{
	mu[1]=1;
	for(int i=2; i<=1e5; i++)
	{
		if(!v[i])
		{
			v[i]=i;
			prime[++tot]=i;
			mu[i]=-1;
		}
			
		for(int j=1; j<=tot; j++)
		{
			if(prime[j]>v[i] || prime[j]>1e5/i)
				break;
			v[i*prime[j]]=prime[j];
			
			if(i%prime[j]==0)
				break;
			mu[i*prime[j]]=-mu[i];
		}
	}
	
	for(int i=1; i<=1e5; i++)
		for(int j=1; i*j<=1e5; j++)
			f[i*j].dat=(f[i*j].dat+i)%MOD;
		
	for(int i=1; i<=1e5; i++)
		f[i].id=i;
		
	sort(f+1,f+1+100000,cmp2);
}

LL calc(int a,int b)
{
	if(a>b)
		swap(a,b);
		
	LL res=0;
	for(int l=1,r; l<=a; l=r+1)
	{
		r=min(a/(a/l),b/(b/l));
		res=(res+(ask(r)-ask(l-1)+MOD)%MOD*1LL*(a/l)%MOD*(b/l)%MOD)%MOD;
	}
	
	return res;
}
 		
int main()
{
	primes();

	scanf("%d",&T);
	for(int i=1; i<=T; i++)
		scanf("%d%d%d",&a[i].n,&a[i].m,&a[i].s),a[i].id=i;
	
	sort(a+1,a+1+T,cmp);
	
	int p=1;
	for(int i=1; i<=T; i++)
	{
		while(f[p].dat<=a[i].s && p<=1e5)
			update(f[p].id,f[p].dat),p++;
		ans[a[i].id]=calc(a[i].n,a[i].m);
	}
	
	for(int i=1; i<=T; i++)
		printf("%lld\n",ans[i]);
    
    return 0;
}