BZOJ 2839: 集合计数 容斥原理 组合数学

2839: 集合计数

Time Limit: 10 Sec  Memory Limit: 128 MB
Submit: 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;
}