2022 ccpc黑龙江省赛E题

题解好像有一些typo，不过问题不大。本题就是经典莫比乌斯反演套路题

关键问题是读入a的时候要把$a$转化成$f(a)$

题意

定义一种函数$f(x)=x的奇数次质因子的乘积，如果x没有奇数次质因子，f(x)=1$

读入$n$个数$b_i$，求${\sum }_{1\le i<j\le n}f(b_i\ast b_j)$
$n,b_i\le 2e5$

题解

首先经过观察可以发现，

$f(b_i\ast b_j)=\frac{f(b_i)\ast f(b_j)}{gc{d}^2(f(b_i),f(b_j))}$

由于是一个和$gcd$有关的求和式子，考虑莫比乌斯反演。

首先枚举$gcd=d$，那么求和式可以变成：
${\sum }_{d=1}^{mx}\frac{1}{d^2}{\sum }_{i=1}^{mx}ton[i]{\sum }_{j=1}^{mx}ton[j]\ast [gcd(i,j)==d]$

这里给出一些定义和转化：
首先读入$b_i$时，将$b_i$转化为$f(b_i)$，不然算$gcd$的贡献时不好处理
$ton[i]表示f(b_j)=i的b_j个数$

然后就把最后的判别式化成${\sum }_{T|gcd(i,j)}\mu (T)$就可以求和了
然后注意处理一下重复被算的部分。

#include<bits/stdc++.h>
using namespace std;
#define int long long
const int mod=1e9+7;
const int maxn=2e5+5,N=2e5;
int mu[maxn],p[maxn],cnt,vis[maxn],f[maxn],b[maxn],n,mx;
void get(){
	mu[1]=1;f[1]=1;
	for(int i=2;i<=N;i++){
		if(!vis[i]){
			p[++cnt]=i;mu[i]=-1;f[i]=i;
		}
		for(int j=1;j<=cnt && i*p[j]<=N;j++){
			vis[i*p[j]]=1; 
			mu[i*p[j]]=-mu[i];
			if(i%p[j]==0){
				vis[i*p[j]]=1;
				mu[i*p[j]]=0;
			}
		}
	}
	f[1]=1;
	for(int i=2;i<=N;i++){
		if(vis[i]){
			//printf("%d\n",i);
			int pos=i;
			f[i]=1;
			for(int j=1;p[j]*p[j]<=pos && j<=cnt;j++){
				int tmp=0;
				while(pos%p[j]==0){
					pos=pos/p[j];
					tmp++;
				}
				if(tmp>0 && (tmp&1)){
					f[i]=f[i]*p[j];
				}
			}
			if(pos!=1){
				f[i]=f[i]*pos;
			}
		}
	}
}
inline int ksm(int a,int b){
	int ans=1;
	while(b){
		if(b&1)ans=ans*a%mod;
		a=a*a%mod;b>>=1;
	}
	return ans;
}
int inv[maxn];
inline int am(int a,int b){
	return a+b>=mod?a+b-mod:a+b;
}
int val[maxn];
signed main(){
	//freopen("E.in","r",stdin);
	//freopen("E.out","w",stdout);
	ios::sync_with_stdio(0);
	cin.tie(0);
	cin>>n;
	mx=0;get();
	for(int i=1;i<=n;i++){
		int a;cin>>a;
		a=f[a];
		mx=max(mx,a);
		b[a]++;
	}
	
	int ans=0;
	int m=n;
	n=mx;
	inv[1]=1;
	for(int i=2;i<=n;i++){
		inv[i]=ksm(i,mod-2);
	}
	for(int i=1;i<=mx;i++){
		if(b[i]>1){
			ans=am(ans,(b[i]*(b[i]-1)/2)%mod);
		}
	}
	for(int d=1;d<=n;d++){
		for(int t=1;t<=n/d;t++){
			int tmp=mu[t];
			if(tmp==-1)tmp=mod-1;
			tmp=tmp*inv[d]%mod*inv[d]%mod;
			int sum=0,alfa=0;
			for(int i=1;i<=n/(t*d);i++){
				int pi=i*t*d;
				sum=am(sum,f[pi]*b[pi]%mod);
				alfa=am(alfa,f[pi]*b[pi]%mod*f[pi]*b[pi]%mod);
			}
			sum=sum*sum%mod;
			sum=am(sum,mod-alfa);
			sum=sum*inv[2]%mod;
			ans=am(ans,tmp*sum%mod);
		}
	}
	printf("%lld\n",ans);
	return 0;
}