2839: 集合计数
Time Limit: 10 Sec Memory Limit: 128 MBSubmit: 521 Solved: 285
[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;
挺水的容斥原理
枚举重合部分的大小设为j就行了
容斥系数是C(j,k)挺好想的
#include<cmath>
#include<ctime>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<iostream>
#include<algorithm>
#include<iomanip>
#include<vector>
#include<string>
#include<bitset>
#include<queue>
#include<set>
#include<map>
using namespace std;
typedef long long ll;
inline int read()
{
int x=0,f=1;char ch=getchar();
while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
while(ch<='9'&&ch>='0'){x=x*10+ch-'0';ch=getchar();}
return x*f;
}
void print(int x)
{if(x<0)putchar('-'),x=-x;if(x>=10)print(x/10);putchar(x%10+'0');}
const int N=1000100,mod=int(1e9)+7;
int fac[N],inv[N];
inline int qpow(int x,int y,int P)
{
int res=1;
while(y)
{
if(y&1)res=1ll*res*x%P;
x=1ll*x*x%P;
y>>=1;
}
return res;
}
inline int C(int n,int m)
{return 1ll*fac[n]*inv[m]%mod*inv[n-m]%mod;}
int main()
{
int n=read(),K=read();
register int i,k,res;
fac[0]=1;for(i=1;i<=n;++i)fac[i]=1ll*fac[i-1]*i%mod;
inv[1]=1;for(i=2;i<=n;++i)inv[i]=1ll*(mod-mod/i)*inv[mod%i]%mod;
inv[0]=1;for(i=1;i<=n;++i)inv[i]=1ll*inv[i-1]*inv[i]%mod;
int ans=0,p=-1;
for(k=K;k<=n;++k)
{
p*=-1;
res=1ll*C(n,k)*C(k,K)%mod*(qpow(2,1ll*qpow(2,n-k,mod-1),mod)-1)%mod;
ans=((ans+p*res)%mod+mod)%mod;
}
cout<<ans<<endl;
return 0;
}