CF1043F Make It One

題意

從一堆數中選擇最少的數，使它們的\(\gcd=1\)

輸入輸出格式

輸入格式

第一行：一個正整數\(n\)。
第二行：\(n\)個正整數，給出了這個數列。

輸出格式

一行，\(-1\)（如果任意選擇都不能得到\(1\)分）或一個正整數（表示選擇的數的數量的最小值）

資料範圍

\(1\leq n\leq 300,000 , 1\leq a_i \leq 300,000\).

第二次遇到此類題了，反演理解的不好，不清楚怎麼用反演理解這個題。

不過容斥的方法還挺神的（雖然可能感覺有點套路？

發現選擇的數不會超過\(7\)個，因為\(2 \times 3 \times 5 \times 7 \times 11 \times 13 \times 17 >3\times 10^5\)

直接暴力列舉選幾個數

然後設\(dp_{i,j}\)表示選\(i\)了\(i\)個數且這\(i\)個數\(\gcd=j\)的方案數

\(dp_{i,j}=\binom{cnt_j}{i}-\sum_{j|k}f_{i,k}\)

\(cnt_j\)表示有多少個數是\(j\)的倍數

Code:

#include <cstdio>
#include <cstring>
#define ll long long
const int N=3e5+10;
const ll mod=1e9+7;
ll qp(ll d,ll k)
{
    ll f=1;
    while(k)
    {
        if(k&1) f=f*d%mod;
        d=d*d%mod;
        k>>=1;
    }
    return f;
}
ll fac[N],inv[N],dp[N];
int n,cnt[N],mx;
int main()
{
    scanf("%d",&n);
    fac[0]=1;
    for(int i=1;i<=n;i++) fac[i]=fac[i-1]*i%mod;
    inv[n]=qp(fac[n],mod-2);
    for(int i=n-1;~i;i--) inv[i]=inv[i+1]*(i+1)%mod;
    for(int a,i=1;i<=n;i++) scanf("%d",&a),++cnt[a],mx=mx>a?mx:a;
    for(int i=1;i<=mx;i++)
        for(int j=i<<1;j<=mx;j+=i)
            cnt[i]+=cnt[j];
    for(int i=1;i<=7;i++)
    {
        memset(dp,0,sizeof(dp));
        for(int j=mx;j;j--)
        {
            dp[j]=cnt[j]>=i?fac[cnt[j]]*inv[cnt[j]-i]%mod*inv[i]%mod:0;
            if(dp[j])
                for(int k=j<<1;k<=mx;k+=j)
                    (dp[j]-=dp[k])%=mod;
            (dp[j]+=mod)%=mod;
        }
        if(dp[1]) return printf("%d\n",i),0;
    }
    puts("-1");
    return 0;
}
 

2018.11.5