省选模拟赛第三场 T2 求和（莫比乌斯反演）

 

 

 

 

 

题解

其实这道题有很多种做法

std题解做法：

我的做法1：

原式：

由于我们之前做过这道题：http://www.51nod.com/Challenge/Problem.html#problemId=1584（加权约数和）

然后就可以直接利用它的结论推到这一步：

考试的时候以为要把后面用线性筛筛出来，然后整除分块（脑抽时间到）

然后发现后面的竟然与前面的k有关，就以为不可做了。。。（其实直接暴力计算就可以了）

时间复杂度O(nlnn)

 

我的做法2：

我发现我推导出来的式子"不可做"，于是就让自己换一个思路

一步步推到这里：（用加权约数和结论的前一步）

不单独处理j/q直接把k循环提出来整理：

设

这两个可以直接暴力预处理

然后整个式子也可以直接计算

好像还比方法1简单一些（少筛一个 ）

 

代码：（做法2）

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
#define N 500005
const int mod=998244353;
int prime[N],mu[N],tot,n,m;
int f[N],g[N];
bool vis[N];
void shai()
{
	vis[1]=1;mu[1]=1;
	int i,j;
	for(i=2;i<=500000;i++){
		if(!vis[i]){
			prime[++tot]=i;
			mu[i]=-1;
		}
		for(j=1;j<=tot;j++){
			int tmp=i*prime[j];
			if(tmp>500000)break;
			vis[tmp]=1;
			if(i%prime[j]==0){
				mu[tmp]=0;
				break;
			}
			mu[tmp]=-mu[i];
		}
	}
	for(i=1;i<=n;i++)
		for(j=i;j<=n;j+=i)
			f[i]=(1ll*f[i]+1ll*i*mu[j]+1ll*mod)%mod;
	for(i=1;i<=m;i++)
		for(j=i;j<=m;j+=i)
			g[i]=(1ll*g[i]+1ll*(j/i)*mu[j]+1ll*mod)%mod;
}
int main()
{
	freopen("sum.in","r",stdin);
	freopen("sum.out","w",stdout);
	int i,j,ans=0,sum1,sum2;
	scanf("%d%d",&n,&m);
	shai();
	for(i=1;i<=n;i++){
		sum1=sum2=0;
		for(j=i;j<=n;j+=i){
			sum1+=f[j];
			if(sum1>=mod)sum1-=mod;
		}
		for(j=i;j<=m;j+=i){
			sum2+=g[j];
			if(sum2>=mod)sum2-=mod;
		}
		ans=(1ll*ans+1ll*mu[i]*sum1%mod*sum2%mod+1ll*mod)%mod;
	}
	printf("%d",(1ll*ans%mod+1ll*mod)%mod);
}