CodeForces 803 F.Coprime Subsequences（莫比乌斯反演+组合数学）

Description

给出一长度为$n$的序列$a_i$，问该序列的子序列中有多少子序列的$gcd$是$1$

Input

第一行输入一整数$n$表示序列长度，之后输入$n$个整数$a_i$表示该序列$(1\le n,a_i\le 10^5)$

Output

输出$a$序列的$gcd$为$1$的子序列，结果模$10^9+7$

Sample Input

3
1 2 3

Sample Output

5

Solution

用$f(d)$表示$a$子序列中$gcd$为$d$的个数，$F(d)$表示$a$子序列中$gcd$被$d$整除的个数

统计$a_1,a_2,...,a_n$中可以被$d$整除的数的个数$num(d)$，则$F(d)=2^{num(d)}-1$

显然$F(d)=\sum\limits_{d|n}f(n)$，由莫比乌斯反演，$f(d)=\sum\limits_{d|n}\mu(\frac{n}{d})F(n)$

而要求的答案即为$f(1)=\sum\limits_{i=1}^m\mu(i)F(i)$，其中$m=max(a_1,a_2,...,a_n)$

Code

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<vector>
#include<queue>
#include<map>
#include<set>
#include<ctime>
using namespace std;
typedef long long ll;
typedef pair<int,int>P;
const int INF=0x3f3f3f3f,maxn=100005;
#define mod 1000000007
int phi[maxn],mu[maxn],p[maxn],f[maxn],n,num[maxn];
void init(int n=1e5)
{
    mu[1]=1;
    int res=0;
    for(int i=2;i<=n;i++)
    {
        if(!phi[i])phi[i]=i-1,mu[i]=-1,p[res++]=i;
        for(int j=0;j<res&&i*p[j]<=n;j++)
        {
            if(i%p[j])
            {
                phi[i*p[j]]=(p[j]-1)*phi[i];
                mu[i*p[j]]=-mu[i];
            }
            else 
            {
                phi[i*p[j]]=p[j]*phi[i];
                mu[i*p[j]]=0;
                break;
            }
        }
    }
    f[0]=1;
    for(int i=1;i<=n;i++)f[i]=2*f[i-1]%mod;
} 
void add(int &x,int y)
{
    x=x+y>=mod?x+y-mod:x+y;
}
int main()
{
    init();
    scanf("%d",&n);
    int m=0;
    for(int i=1;i<=n;i++)
    {
        int temp;
        scanf("%d",&temp);
        m=max(m,temp);
        num[temp]++;
    }
    for(int i=1;i<=m;i++)
        for(int j=2*i;j<=m;j+=i)
            num[i]+=num[j];
    int ans=0;
    for(int i=1;i<=m;i++)
        if(mu[i]==1)add(ans,f[num[i]]-1);
        else if(mu[i]==-1)add(ans,mod-(f[num[i]]-1));   
    printf("%d\n",ans);
    return 0;
}