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

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

∑i=0xf(i)Cix∑i=0xf(i)Cxi

我们设有一个函数 g(x)g(x)，

g(x)={x=kx≠k10g(x)={x=k1x≠k0

则

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

根据 二项式反演，

f(x)=∑i=0x(−1)x−iCixg(x)f(x)=∑i=0x(−1)x−iCxig(x)

考虑到仅有 g(k)=1g(k)=1,
f(x)={0(−1)(x−k)Ckxx<kx≥kf(x)={0x<k(−1)(x−k)Cxkx≥k
因此答案为
ansans
=∑ni=0f(i)α(i)=∑i=0nf(i)α(i)
=∑ni=k(−1)(i−k)CkiCin×(22n−i−1)=∑i=kn(−1)(i−k)CikCni×(22n−i−1)
这个东西可以直接 O(n)O(n)求出，线性处理一下阶乘逆元即可。而且注意 22t=(22t−1)222t=(22t−1)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;
}

转载于:https://www.cnblogs.com/2016gdgzoi471/p/9476846.html