Label
容斥原理+Möbius函数解决数论相关问题
Description
多组询问:给定 n n n个数 a i ( n , a i ≤ 1 0 4 ) a_i(n,a_i\leq 10^4) ai(n,ai≤104),试求 ∑ a , b , c , d ∈ a [ g c d ( a , b , c , d ) = = 1 ] \sum_{a,b,c,d\in a}[gcd(a,b,c,d)==1] ∑a,b,c,d∈a[gcd(a,b,c,d)==1]
Solution
有了解决p3455的基础,我们不难推导此题的解法。
设
d
i
d_i
di为
a
a
a中被
d
d
d整除的数的个数,我们利用类似埃氏筛的方法求出
d
1
∼
d
n
d_1\sim d_n
d1∼dn,接下来直接套用p3455的容斥思路与公式可得:
a
n
s
=
∑
i
=
1
m
a
x
{
a
i
}
μ
(
i
)
C
d
i
4
ans=\sum_{i=1}^{max\{a_i\}}\mu(i)C_{d_i}^{4}
ans=i=1∑max{ai}μ(i)Cdi4
Code
#include<cstdio>
#include<cstring>
#include<iostream>
#define ri register int
#define ll long long
using namespace std;
const int MAXN=1e4;
int n,miu[MAXN+20],prime[MAXN],cnt,a[MAXN+20],maxn,scnt[MAXN+20];
ll ans;
bool notprime[MAXN+20],vis[MAXN+20];
void Mobius()
{
miu[1]=1,notprime[1]=true;
for(ri i=2;i<=MAXN;++i)
{
if(!notprime[i]) prime[++cnt]=i,miu[i]=-1;
for(ri j=1;j<=cnt&&i*prime[j]<=MAXN;++j)
{
notprime[i*prime[j]]=true;
if(i%prime[j]==0) break;
else miu[i*prime[j]]=-miu[i];
}
}
}
int main()
{
Mobius();
while(~scanf("%d",&n))
{
memset(scnt,0,sizeof(scnt));
memset(vis,0,sizeof(vis));
maxn=0,ans=0LL;
for(ri i=1;i<=n;++i)
{
scanf("%d",&a[i]);
maxn=max(maxn,a[i]); vis[a[i]]=1;
}
scnt[1]=n;
for(ri i=2;i<=maxn;++i)
for(ri j=1;i*j<=maxn;++j)
if(vis[i*j]) ++scnt[i];
for(ri i=1;i<=maxn;++i)
if(scnt[i]>=4) ans+=(ll)miu[i]*(ll)(scnt[i]-3)*(ll)(scnt[i]-2)*(ll)(scnt[i]-1)*(ll)scnt[i]/24LL;
cout<<ans<<'\n';
}
return 0;
}