洛谷 P2714 四元组统计 题解

题目传送门

题目大意： 求有多少个四元组满足 $\gcd (a_i,a_j,a_k,a_l)=1$。

题解

看到 $\gcd$ 自然就想到了莫比乌斯反演。

于是设 $f(d)$ 表示满足 $\gcd (a_i,a_j,a_k,a_l)=d$ 的四元组数，$F(n)$ 表示满足 $n|\gcd (a_i,a_j,a_k,a_l)$ 的四元组数。

显然有：
$$F(n)=\sum _{n|d}f(d)$$

反演一下得到：
$$f(d)=\sum _{d|n}F(n)\mu (\frac{n}{d})$$

显然答案是 $f(1)$，于是直接让 $d=1$，代入得（这下面的 $n$ 和上面的 $n$ 的意义不一样，这下面的 $n$ 的意义是序列中的最大值）：
$$f(1)=\sum _{i=1}^{n}F(i)\mu (i)$$

$F$ 也不难求，假如统计一个 $tot$，$to{t}_i$ 表示序列中有多少个 $i$ 的倍数，那么有 $F(i)=C_{to{t}_i}^4$。

然而 $tot$ 也很好求，用类似埃氏筛的方法就能搞出来。

具体见代码吧：

#include <cstdio>
#include <cstring>
#define maxn 10010
#define ll long long

int prime[maxn],t=0,mu[maxn];
bool v[maxn];
int n,a[maxn],b[maxn],tot[maxn];
void work()//筛mu
{
	mu[1]=1;
	for(int i=2;i<=maxn-10;i++)
	{
		if(!v[i])prime[++t]=i,mu[i]=-1;
		for(int j=1;j<=t&&i*prime[j]<=maxn-10;j++)
		{
			v[i*prime[j]]=true;
			if(i%prime[j]==0)break;
			mu[i*prime[j]]=-mu[i];
		}
	}
}
inline int max(int x,int y){return x>y?x:y;}
ll C(int x)//求 C(x,4)
{
	if(x<4)return 0;
	return (ll)x*(x-1)*(x-2)*(x-3)/2ll/3ll/4ll;
}

int main()
{
	work();
	while(~scanf("%d",&n))
	{
		memset(b,0,sizeof(b));
		memset(tot,0,sizeof(tot));
		int ma=0;
		for(int i=1;i<=n;i++)
		scanf("%d",&a[i]),b[a[i]]++,ma=max(a[i],ma);
		for(int i=1;i<=ma;i++)
		for(int j=1;i*j<=ma;j++)
		tot[i]+=b[i*j];//求tot[i]
		ll ans=0;
		for(int i=1;i<=ma;i++)
		ans+=C(tot[i])*(ll)mu[i];//统计答案
		printf("%lld\n",ans);
	}
}