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;
}