bzoj 2839 集合计数

2839: 集合计数
Time Limit: 10 Sec Memory Limit: 128 MB
Submit: 414 Solved: 228
[Submit][Status][Discuss]
Description

一个有N个元素的集合有2^N个不同子集（包含空集），现在要在这2^N个集合中取出若干集合（至少一个），使得

它们的交集的元素个数为K，求取法的方案数，答案模1000000007。（是质数喔~）

Input

一行两个整数N,K

Output

一行为答案。

Sample Input

3 2

Sample Output

6

HINT

【样例说明】

假设原集合为{A,B,C}

则满足条件的方案为：{AB,ABC},{AC,ABC},{BC,ABC},{AB},{AC},{BC}

【数据说明】

 对于100%的数据，1≤N≤1000000；0≤K≤N；

Source

---

【分析】
容斥原理
参见阿当学长的博客…实在不想写数学公式…累
http://blog.csdn.net/aarongzk/article/details/51520454
顺便提一句，题解中的
$ans=∑f[i]*C(i,k)*(-1)^{i-k}，k≤i≤n。$
应该改为
$ans=∑f[i]*C(n,i)*C(i,k)*(-1)^{i-k}，k≤i≤n。$

看了一个半小时题解没看懂
结果代码和题解不一样QaQ，学长要敬业啊QaQQQ

---

【代码】

//bzoj 2839 集合计数
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#define ll long long
#define M(a) memset(a,0,sizeof a)
#define fo(i,j,k) for(i=j;i<=k;i++)
using namespace std;
const int mod=1e9+7;
const int mxn=1000005;
int n,m,k;
ll ans,tmp;
int fac[mxn],inv[mxn];
inline void init()
{
    int i,j;
    fac[0]=inv[0]=inv[1]=1;
    fo(i,1,n) fac[i]=(ll)fac[i-1]*i%mod;
    fo(i,2,n) inv[i]=(ll)(mod-mod/i)*inv[mod%i]%mod;
    fo(i,1,n) inv[i]=(ll)inv[i]*inv[i-1]%mod;
}
int main()
{
    int i,j;
    scanf("%d%d",&n,&k);
    init();
    ll x=2;
    for(i=n;i>=k;i--)
    {
        if(i!=n) x=x*x%mod;
        tmp=(ll)fac[n]*inv[n-i]%mod*inv[k]%mod*inv[i-k]%mod;
        if((i-k)&1) tmp=tmp*(mod-1)%mod;
        ans=(ans+(x+mod-1)%mod*tmp%mod)%mod;
    }
    printf("%lld\n",ans);
    return 0;
}