P3911最小公倍数之和 题解

2021-7-31

---

这道题使用莫比乌斯反演，比较套路化

P3911

求${\sum }_{i=1}^n{\sum }_{j=1}^{n}lcm(A_i,A_j)$。

可以发现$A_i$的值不算特别大，瞬间想到桶（剧透）

我们定义$a_i$为数$i$的出现次数，$m$为一个值出现最多的次数。

那么开始推式子：

$$Ans=\sum _{i=1}^m\sum _{j=1}^m{a}_i{a}_j\times lcm(i,j)$$
根据小学奥数知识（真就小奥）的$a\times b=gcd(a,b)\times lcm(a,b)$

可得：

$$Ans=\sum _{d=1}^m\sum _{i=1}^m\sum _{j=1}^m\frac{ij{a}_i{a}_j}{d}[gcd(i,j)=d]$$

$$Ans=\sum _{d=1}^m\frac{1}{d}\sum _{i=1}^m\sum _{j=1}^{m}ij{a}_i{a}_j[gcd(i,j)=d]$$

令 $$f(x)=\sum _{i=1}^m\sum _{j=1}^{m}ij{a}_i{a}_j[gcd(i,j)=d]$$

构造 $$F(x)=\sum _{x|d}f(d)$$

可得

$$F(x)=\sum _{i=1}^m\sum _{j=1}^{m}ij{a}_i{a}_j[x|gcd(i,j)]$$
$$=\sum _{x|i}\sum _{x|j}ij{a}_i{a}_j$$
$$=\sum _{x|i}i{a}_i\times \sum _{x|j}j{a}_j$$
$$=(\sum _{x|i}i{a}_i{)}^2$$

由反演可得：

$$f(d)=\sum _{x=1}^{\lfloor \frac{m}{d}\rfloor }\mu (x)F(dx)$$

那么带入$Ans$

$$Ans=\sum _{d=1}^m\frac{1}{d}f(d)$$
$$=\sum _{d=1}^m\frac{1}{d}\sum _{g=1}^{\lfloor \frac{m}{d}\rfloor }\mu (g)F(gd)$$

我们可以用$O(n\log n)$的时间复杂度求出$F$，然后用$O(n\log n)$的时间复杂度求出该式子

代码：

#include<bits/stdc++.h>
#define inf 0x3f3f3f3f
#define ll long long
using namespace std;
#define int long long
const int Maxn=50010;
int n,Ar[Maxn];
int m,Tong[Maxn];
int f[Maxn],ans;
int ret,prime[Maxn],mu[Maxn];
bool flag[Maxn];
void get_mu()
{
	mu[1]=1;
	for(int i=2;i<=m;i++)
	{
		if(!flag[i])
		{
			prime[++ret]=i;
			mu[i]=-1;
		}
		for(int j=1;j<=ret && prime[j]<=m/i;j++)
		{
			flag[i*prime[j]]=true;
			if(i%prime[j]==0)
			{
				mu[i*prime[j]]=0;
				break;
			}else{
				mu[i*prime[j]]=-mu[i];
			}
		}
	}
}
signed main()
{
	
	scanf("%lld",&n);
	for(int i=1;i<=n;i++)
	{
		scanf("%lld",&Ar[i]);
		m=max(m,Ar[i]);
		Tong[Ar[i]]++;
	}
	//m=50000;
	for(int i=1;i<=m;i++)
	{
		for(int j=i;j<=m;j+=i)
		{
			f[i]+=Tong[j]*j;
		}
	}
	for(int i=1;i<=m;i++)
	{
		f[i]*=f[i];
	}
	get_mu();
	for(int i=1;i<=m;i++)
	{
		int tot=0;
		for(int j=1;j<=m/i;j++)
		{
			tot+=mu[j]*f[i*j];
		}
		ans+=tot/i;
	}
	printf("%lld\n",ans);
	return 0;
}

然而还有一道比较相似的题目：

[AGC038C] LCMs

只不过是把问题改为了：

$$\sum _{i=1}^n\sum _{j=i+1}^{n}lcm(A_i,A_j)$$的值模$998244353$

然而式子也比较像。

双倍经验！！！111

代码：

#include<bits/stdc++.h>
#define inf 0x3f3f3f3f
#define ll long long
using namespace std;
const int Mod=998244353;
const int inv2=499122177;
#define int long long
const int Maxn=1e6+10;
int n,Ar[Maxn];
int m,Tong[Maxn];
int f[Maxn],ans;
int ret,prime[Maxn],mu[Maxn];
bool flag[Maxn];
void get_mu()
{
	mu[1]=1;
	for(int i=2;i<=m;i++)
	{
		if(!flag[i])
		{
			prime[++ret]=i;
			mu[i]=-1;
		}
		for(int j=1;j<=ret && prime[j]<=m/i;j++)
		{
			flag[i*prime[j]]=true;
			if(i%prime[j]==0)
			{
				mu[i*prime[j]]=0;
				break;
			}else{
				mu[i*prime[j]]=-mu[i];
			}
		}
	}
}
signed main()
{
	
	scanf("%lld",&n);
	for(int i=1;i<=n;i++)
	{
		scanf("%lld",&Ar[i]);
		m=max(m,Ar[i]);
		Tong[Ar[i]]++;
		ans-=Ar[i];
	}
	//m=50000;
	get_mu();
	for(int i=1;i<=m;i++)
	{
		for(int j=i;j<=m;j+=i)
		{
			f[j]=(f[j]+mu[i]*i%Mod);
		}
	}
	for(int i=1;i<=m;i++)
	{
		int tot=0;
		for(int j=1;j<=m/i;j++)
		{
			tot=(tot+j*Tong[i*j]%Mod)%Mod;
		}
		ans=(ans+i*f[i]%Mod*tot%Mod*tot%Mod)%Mod;
	}
    ans=(ans*inv2%Mod+Mod)%Mod;
	printf("%lld\n",ans);
	return 0;
}

莫比乌斯反演大法好！！！11