洛谷 CF1043F Make It One 解题报告

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

猜你喜欢

转载自www.cnblogs.com/ppprseter/p/9911563.html