Codeforces803 F.Coprime Subsequences（容斥）

最新推荐文章于 2022-07-14 18:49:43 发布

live4m

最新推荐文章于 2022-07-14 18:49:43 发布

阅读量224

点赞数

本文链接：https://blog.csdn.net/weixin_44178736/article/details/108925929

版权

该博客讨论了一种计算给定序列中gcd为1的子序列数量的算法。通过容斥原理，先计算每个数i作为gcd的子序列数量，然后逐次排除i的倍数，最终得到gcd为1的子序列数量。代码实现中，使用了前缀和和动态规划的方法，对每个数的倍数进行统计和更新。

摘要由CSDN通过智能技术生成

题意：

给定长度为n的序列，
问有多少个gcd为1的子序列。
答案对1e9+7取模。

数据范围：n<=1e5，a(i)<=1e5

解法：

$设 f (i) 为 g c d 为 i 的子序列数量$

$考虑容斥, 先计算出 g c d = i 的倍数的数量, 然后容斥掉 g c d 为 i * 2, i * 3 . . 的数量$

$设cnt[i]为序列中i的数量,那么f[i]=(2^{cnt[i]}-1)-f[i*2]-f[i*3]...$

$其中2^{cnt[i]-1}是从所有i的倍数中选出若干个数的方案数,$

$即C_{cnt[i]}^1+C_{cnt[i]}^2..C_{cnt[i]}^{cnt[i]}，因为不能取空集，因此减掉C_{cnt[i]}^0,$

$从后往前逆推即可$

code：

#include<bits/stdc++.h>
using namespace std;
#define int long long
const int maxm=1e5+5;
const int mod=1e9+7;
int f[maxm];
int a[maxm];
int cnt[maxm];
int p2[maxm];
int n;
signed main(){
    p2[0]=1;
    for(int i=1;i<maxm;i++)p2[i]=p2[i-1]*2%mod;
    cin>>n;
    for(int i=1;i<=n;i++)cin>>a[i],cnt[a[i]]++;
    for(int i=1;i<maxm;i++){
        for(int j=i+i;j<maxm;j+=i){
            cnt[i]+=cnt[j];
        }
    }
    for(int i=maxm-1;i>=1;i--){
        f[i]=p2[cnt[i]]-1;
        for(int j=i+i;j<maxm;j+=i){
            f[i]=(f[i]-f[j])%mod;
        }
    }
    cout<<(f[1]%mod+mod)%mod<<endl;
    return 0;
}