题目
题目链接:https://codeforces.com/problemset/problem/585/E
有一个包含 \(n\) 个 \(\in [2, 10^7]\) 的整数的可重集合。
要求满足条件的一个元素 \(x\) 和一个集合 \(S\) 的方案数。
条件:\(x \notin S\),\(\gcd\{S\} > 1\),\(\gcd(x, \gcd\{S\}) = 1\)。
\(n \le 5 \times 10^5\),元素 \(\leq 10^7\),答案对 \(10^9+7\) 取模。
思路
考虑求出 \(f_i\) 表示集合中所有数字中与 \(i\) 互质的数字数量。令 \(c_i\) 表示数字 \(i\) 出现的次数。
那么
后面那个玩意可以用 Dirichlet 后缀和搞出来,然后再 Dirichlet 前缀和一次就可以求出 \(f\)。
然后考虑求 \(h_i\) 表示选出若干个数,\(\gcd=i\) 的方案数。\(h'_i\) 表示选出若干个数 \(gcd\) 是 \(i\) 的倍数的方案数。那么显然有
那么直接把 Dirichlet 前缀和倒序变为差分即可。
最终答案就是
时间复杂度 \(O(n\log\log n)\)。
代码
#include <bits/stdc++.h>
using namespace std;
const int N=10000010,MOD=1e9+7;
int n,m,ans,f[N],g[N],h[N],mu[N],prm[N];
bool v[N];
void findprm(int n)
{
mu[1]=1;
for (int i=2;i<=n;i++)
{
if (!v[i]) prm[++m]=i,mu[i]=-1;
for (int j=1;j<=m;j++)
{
if (i>n/prm[j]) break;
v[i*prm[j]]=1; mu[i*prm[j]]=-mu[i];
if (!(i%prm[j])) { mu[i*prm[j]]=0; break; }
}
}
}
int fpow(int x,int k)
{
int ans=1;
for (;k;k>>=1,x=1LL*x*x%MOD)
if (k&1) ans=1LL*ans*x%MOD;
return ans;
}
int main()
{
scanf("%d",&n);
for (int i=1,x;i<=n;i++)
{
scanf("%d",&x);
g[x]++; m=max(m,x);
}
n=m; m=0;
findprm(n);
for (int i=1;i<=m;i++)
for (int j=n/prm[i];j>=1;j--)
g[j]=(g[j]+g[j*prm[i]])%MOD;
for (int i=1;i<=n;i++)
h[i]=fpow(2,g[i])-1;
for (int i=1;i<=n;i++)
f[i]=(g[i]*mu[i]+MOD)%MOD;
for (int i=1;i<=m;i++)
for (int j=1;j*prm[i]<=n;j++)
f[j*prm[i]]=(f[j*prm[i]]+f[j])%MOD;
for (int i=1;i<=m;i++)
for (int j=1;j*prm[i]<=n;j++)
h[j]=(h[j]-h[j*prm[i]]+MOD)%MOD;
for (int i=2;i<=n;i++)
ans=(ans+1LL*f[i]*h[i])%MOD;
printf("%d",ans);
return 0;
}