BZOJ2820 YY的gcd【莫比乌斯反演】【分块】

最新推荐文章于 2022-01-28 15:38:33 发布

DenyTian

最新推荐文章于 2022-01-28 15:38:33 发布

阅读量180

点赞数

分类专栏：【解题报告】 ====莫比乌斯反演

本文链接：https://blog.csdn.net/qq_34896694/article/details/87374450

版权

【解题报告】同时被 2 个专栏收录

82 篇文章 0 订阅

订阅专栏

====莫比乌斯反演

3 篇文章 0 订阅

订阅专栏

题目链接：https://www.lydsy.com/JudgeOnline/problem.php?id=2820

题意：
求二元组 $(x, y)$ 的个数，满足 $\le x \le n，1 \le y \le m$ ，且 $g c d (x, y)$ 为素数

题解：
令 $P$ 为素数集合，以下除法均为整除
$\large \Sigma_{p \in P}\Sigma_{i=1}^n\Sigma_{j=1}^m [gcd(i,j)==p]$
$\large =\Sigma_{p \in P}\Sigma_{i=1}^{n/p}\Sigma_{j=1}^{m/p} [gcd(i,j)==1]$
$\mu$ 可得
$\large =\Sigma_{p \in P}\Sigma_{i=1}^{n/p}\Sigma_{j=1}^{m/p} \Sigma_{d|x,d|y}\mu(d)$
枚举 $d$ ，交换枚举顺序可得：
$\large =\Sigma_{p \in P}\Sigma_{d=1}^{min(n,m)\over p}\mu(d)[{n\over pd}][{m\over pd}]$
$\large =\Sigma_{t=1}^{min(n,m)}[{n\over t}][{m\over t}]\Sigma_{p \in P,p|t}\mu({t\over p})$
前部分分块即可，后部分我们继续考虑反演
令后半部分为 $g (n)$ ,即 $g(n)=\Sigma_{p \in P,p|n}\mu({n \over p})$
$\large =\Sigma_{t=1}^{min(n,m)}[{n\over t}][{m\over t}]g(t)$
$g (t)$ 可以通过前缀和算出来（分块的话不是一个 $t$ ）

代码：

// by Balloons
#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
#define mpr make_pair
#define debug() puts("okkkkkkkk")
#define rep(i,a,b) for(int (i)=(a);(i)<=(b);(i)++)

using namespace std;

typedef long long LL;

const int inf = 1 << 30;
const int maxn=1e7+5;

int n,m;
int notpm[maxn],pm[maxn],pcnt=0;
int mu[maxn];
LL cnt[maxn]; 

void xxs(){
	notpm[1]=mu[1]=1;
	for(int i=2;i<=maxn-5;i++){
		if(!notpm[i]){
			pm[++pcnt]=i;mu[i]=-1;
		}
		for(int j=1;j<=pcnt && pm[j]*i<=maxn-5; j++){
			notpm[i*pm[j]]=1;
			if(i%pm[j] == 0){
				mu[i*pm[j]] = 0;
				break;
			}
			mu[i*pm[j]]=-mu[i];
		}
	}
}

void init(){
	for(int i=1;i<=pcnt;i++){
		for(int j=pm[i];j<=maxn - 5;j+=pm[i])cnt[j] += 1ll*mu[j/pm[i]]; 
	} 
}

void fk(){
	int i=1,j=0;
	LL ans=0;
	for(;i<=n && i<=m; i=j+1){
		j=min(n/(n/i),m/(m/i));
		ans+=1ll*(n/i)*(m/i)*(cnt[j]-cnt[i-1]);
	}
	printf("%lld\n",ans);
}

int main(){
	int Te;scanf("%d",&Te);
	xxs();init();
	for(int i=1;i<=maxn-5;i++)cnt[i]+=cnt[i-1];
	while(Te--){
		scanf("%d%d",&n,&m);
		fk();
	}

	return 0;
}