[Luogu P1829] [国家集训队]Crash的数字表格 / JZPTAB (莫比乌斯反演)

题面

传送门:洛咕


Solution

调到自闭,我好菜啊
为了方便讨论,以下式子\(m>=n\)
为了方便书写,以下式子中的除号均为向下取整

我们来颓柿子吧qwq
显然,题目让我们求:
\(\large \sum_{i=1}^n\sum_{j=1}^m lcm(i,j)\)

\(lcm\)没法玩,我们转到\(gcd\)形式:
\(\large \sum_{i=1}^n\sum_{j=1}^m \frac{i*j}{gcd(i,j)}\)

根据套路,我们去枚举\(gcd\)
\(\large \sum_{i=1}^n\sum_{j=1}^m\sum_{d=1}^{n} \frac{i*j}{d}[gcd(i,j)=d]\)
然后可以把\(d\)的和号移到前面去
\(\large \sum_{d=1}^{n}\sum_{i=1}^n\sum_{j=1}^m \frac{i*j}{d}[gcd(i,j)=d]\)

要让\(gcd(i,j)=d\)\(i,j\)都必须要为\(d\)的倍数,我们可以将原来的\(i*d,j*d\)映射为\(i,j\),有:
\(\large \sum_{d=1}^{n}\sum_{i=1}^{n/d}\sum_{j=1}^{m/d} {i*j}*d[gcd(i,j)=1]\)
\(d\)移到前面去
\(\large \sum_{d=1}^{n}d\sum_{i=1}^{n/d}\sum_{j=1}^{m/d} {i*j}[gcd(i,j)=1]\)

然后我们可以套路地根据\([x=1]=\sum_{d|x}\mu(d)\)这个柿子把\(gcd(i,j)\)处理掉:
\(\large \sum_{d=1}^{n}d\sum_{i=1}^{n/d}\sum_{j=1}^{m/d} {i*j}\sum_{k|gcd(i,j)}\mu(k)\)

根据套路,我们把这种奇奇怪怪的和式变为枚举的形式
\(\large \sum_{d=1}^{n}d\sum_{i=1}^{n/d}\sum_{j=1}^{m/d} {i*j}\sum_{k=1}^{n/d}[k|gcd(i,j)]\mu(k)\)
然后就可以把\(k\)往前提了
\(\large \sum_{d=1}^{n}d\sum_{k=1}^{n/d}\sum_{i=1}^{n/d}\sum_{j=1}^{m/d} {i*j}*[k|gcd(i,j)]\mu(k)\)
要有\(k|gcd(i,j)\)\(i,j\)一定要为\(k\)的倍数
\(\large \sum_{d=1}^{n}d\sum_{k=1}^{n/d}\sum_{i=1}^{\frac{n}{d*k}}\sum_{j=1}^{\frac{m}{d*k}} {i*j*k^2}*\mu(k)\)

然后我们简单的移一下项方便处理
\(\large \sum_{d=1}^{n}d\sum_{k=1}^{n/d}*\mu(k)*k^2\sum_{i=1}^{\frac{n}{d*k}}i\sum_{j=1}^{\frac{m}{d*k}} j\)
后面的\(j\)\(i\)没有半毛钱关系,我们可以把它分离开来
\(\large \sum_{d=1}^{n}d\sum_{k=1}^{n/d}*\mu(k)*k^2(\sum_{i=1}^{\frac{n}{d*k}}i)(\sum_{j=1}^{\frac{m}{d*k}} j)\)

搞定,到这里为止,我们所有东西都可以求了。
对于前面的\(d\)的和式,我们可以发现当\(n/d,m/d\)不变的时候,后面的柿子计算出来的结果是一样的,因此我们可以\(O(\sqrt n)\)来整除分块掉前面那个和式。
后面的那个柿子我们可以再来一次整数除法来计算:最后面的两个和式都是等差数列,前面的\(\mu(k)*k^2\)可以前缀和直接计算。

总复杂度\(O(\sqrt n * \sqrt n)=O(n)\)

但是这题还有一个\(O(\sqrt n)\)的做法,蒟蒻太菜了不会,就不说了


Code

这题细节繁多,请注意多膜以防乘爆
预处理中的\(i^2\)会爆int,请注意

//Luogu  P1829 [国家集训队]Crash的数字表格 / JZPTAB
//Jan,23rd,2019
//莫比乌斯反演
#include<iostream>
#include<cstdio>
using namespace std;
long long read()
{
	long long x=0,f=1; char c=getchar();
	while(!isdigit(c)){if(c=='-') f=-1;c=getchar();}
	while(isdigit(c)){x=x*10+c-'0';c=getchar();}
	return x*f;
}
const int N=10000000+1000;
const int M=10000000;
const int poi=20101009;
int prime[N],cnt_p,mu[N];
bool noPrime[N];
void GetPrime(int n)
{
	noPrime[1]=true,mu[1]=1;
	for(int i=2;i<=n;i++)
	{
		if(noPrime[i]==false)
			prime[++cnt_p]=i,mu[i]=-1;
		for(int j=1;j<=cnt_p and i*prime[j]<=n;j++)
		{
			noPrime[i*prime[j]]=true;
			if(i%prime[j]==0)
			{
				mu[i*prime[j]]=0;
				break;
			}
			mu[i*prime[j]]=mu[i]*mu[prime[j]];
		}
	}
}
long long n,m,pre_mu[N];
long long f(int d)
{
	long long t_ans=0;
	for(long long l=1,r;l<=n/d;l=r+1)
	{
		r=min((n/d)/((n/d)/l),(m/d)/((m/d)/l));
		t_ans=(t_ans+(pre_mu[r]-pre_mu[l-1])*(((1+n/d/l)*(n/d/l)/2)%poi)%poi*(((1+m/d/l)*(m/d/l)/2)%poi))%poi;
	}
	return (t_ans%poi+poi)%poi;
}
int main()
{
	n=read(),m=read();
	if(n>m) swap(n,m);
	
	GetPrime(m);
	for(long long i=1;i<=m;i++)
		pre_mu[i]=((pre_mu[i-1]+mu[i]*i*i)%poi+poi)%poi;
	long long ans=0;
	for(long long l=1,r;l<=n;l=r+1)
	{
		r=min(n/(n/l),m/(m/l));
		ans=((ans+(l+r)*(r-l+1)/2%poi*f(l))%poi+poi)%poi;
	}
	
	printf("%lld",ans);
	return 0;
}

posted @ 2019-01-24 10:04  GoldenPotato  阅读(227)  评论(0编辑  收藏  举报