莫比乌斯反演例题

tanjunming2020

于 2023-02-08 19:39:32 发布

阅读量1.2k

点赞数 6

分类专栏：题解文章标签： c++

本文链接：https://blog.csdn.net/tanjunming2020/article/details/128941997

版权

题解专栏收录该内容

348 篇文章 0 订阅

订阅专栏

题目描述

给你正整数 $a, b, c$ ，求

$\sum\limits_{i=1}^a\sum\limits_{j=1}^b\sum\limits_{k=1}^cf(i,j,k)$

其中 $f$ 函数的定义如下：

$\left\{\begin{matrix} 1\qquad \gcd(i,j,k)=1 \\ \\ 0\qquad \gcd(i,j,k)\neq 1 \end{matrix}\right.$

输出答案模 $10^9+7$ 后的值。

输入格式

一行三个正整数 $a, b, c$ 。

输出格式

一行一个正整数，表示答案。

输入样例1

3 4 5

输出样例1

输入样例2

816 816 816

输出样例2

数据范围

$a\times b\times c\leq 4\times 10^{18}$

题解

前置知识：莫比乌斯反演

我们不妨设 $a\leq b\leq c$ ，则显然 $1\leq a\leq 2\times 10^6$ 。

题目即求

$\sum\limits_{i=1}^a\sum\limits_{j=1}^b\sum\limits_{k=1}^c[\gcd(i,j,k)==1]$

利用莫比乌斯函数的性质，原式变为

$\sum\limits_{i=1}^a\sum\limits_{j=1}^b\sum\limits_{k=1}^c\sum\limits_{d|i,d|j,d|k}\mu(d)$

枚举 $d$ 得

$\sum\limits_{d=1}^a\mu(d)\times \lfloor\dfrac ad\rfloor\times \lfloor\dfrac bd\rfloor\times \lfloor\dfrac cd\rfloor$

预处理出 $\mu$ 函数即可，时间复杂度为 $O (a)$ 。

code

#include<bits/stdc++.h>
#define N 2000000
using namespace std;
int z[N+5],p[N+5],mu[N+5];
long long a,b,c,ans;
long long mod=1000000007;
void init(){
	mu[1]=1;
	for(int i=2;i<=N;i++){
		if(!z[i]){
			p[++p[0]]=i;mu[i]=-1;
		}
		for(int j=1;j<=p[0]&&i*p[j]<=N;j++){
			z[i*p[j]]=1;
			if(i%p[j]==0) break;
			mu[i*p[j]]=-mu[i];
		}
	}
}
int main()
{
	init();
	scanf("%lld%lld%lld",&a,&b,&c);
	if(b>c) swap(b,c);
	if(a>b) swap(a,b);
	if(b>c) swap(b,c);
	for(int i=1;i<=a;i++){
		ans=(ans+1ll*mu[i]*(a/i)%mod*(b/i)%mod*(c/i)%mod+mod)%mod;
	}
	printf("%lld",ans);
	return 0;
}