【BZOJ2839】集合计数 【容斥原理】【二项式反演】

题意:一个有n个元素的集合有2n个不同子集(包含空集),现在要在这2N个集合中取出若干集合(至少一个),使得它们的交集的元素个数为k,求取法的方案数,答案模109+7
题解:我们先考虑有多少种取法,使得这些集合的交的大小至少为k。答案记为α(k)。显然α(k)=Cnk×(22nk1),相当于钦定k个元素每个集合都有,剩下的组成一堆集族,每个都可以选或者不选。这里的减一是因为集族中不能一个都不选,否则就没有意义了。
接下来我们考虑构造一个容斥系数f,使得ans=i=0nf(i)α(i)
我们考虑每一种交集为x的选法,它被统计到的贡献。显然,交集为x的一种选法会被交集至少为1..x的统计到。如果交集为x的一种选法被交集至少为i的统计到了,说明交集为x的这一种选法有i个被钦定了,剩下仍然没有选的一些集族只有唯一的一种组合方式可以拼出这一种交集为x的选法。
因此,每一种每一种交集为x的选法被统计到的贡献为

i=0xf(i)Cxi

我们设有一个函数g(x)
g(x)={x=k1xk0


g(x)=i=0xf(i)Cxi

根据二项式反演
f(x)=i=0x(1)xiCxig(x)

考虑到仅有g(k)=1,
f(x)={0x<k(1)(xk)Cxkx≥k
因此答案为
ans
=i=0nf(i)α(i)
=i=kn(1)(ik)CikCni×(22ni1)
这个东西可以直接O(n)求出,线性处理一下阶乘逆元即可。而且注意22t=(22t1)2
代码:

#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
using namespace std;
const int N=1000005,mod=1000000007;
int n,k,ans,jc[N],inv[N];
int fastpow(int a,int x){
    int res=1;
    while(x){
        if(x&1){
            res=1LL*res*a%mod;
        }
        x>>=1;
        a=1LL*a*a%mod;
    }
    return res;
}
int C(int n,int m){
    return 1LL*jc[n]*inv[m]%mod*inv[n-m]%mod;
}
int main(){
    scanf("%d%d",&n,&k);
    jc[0]=1;
    for(int i=1;i<=n;i++){
        jc[i]=1LL*jc[i-1]*i%mod;
    }
    inv[n]=fastpow(jc[n],mod-2);
    for(int i=n-1;i>=0;i--){
        inv[i]=1LL*inv[i+1]*(i+1)%mod;
    }
    for(int i=n,base=2;i>=k;i--){
        if((i-k)&1){
            ans=(ans-1LL*C(i,k)*C(n,i)%mod*(base-1)%mod+mod)%mod;
        }else{
            ans=(ans+1LL*C(i,k)*C(n,i)%mod*(base-1))%mod;
        }
        base=1LL*base*base%mod;
    }
    printf("%d\n",ans);
    return 0;
}
posted @ 2018-08-04 08:46  一剑霜寒十四洲  阅读(178)  评论(0编辑  收藏  举报