Bzoj2839 集合计数

C. 集合计数

题目描述

一个有N个元素的集合有2^N个不同子集(包含空集),现在要在这2^N个集合中取出若干集合(至少一个),使得它们的交集的元素个数为K,求取法的方案数,答案模1000000007。(是质数喔~)

输入格式

一行两个整数N,K

输出格式

一行为答案。

样例

样例输入

3 2

样例输出

6

数据范围与提示

样例说明

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

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

数据说明

对于100%的数据,1≤N≤1000000;0≤K≤N;

 ab%c!=ab%c!!!卡了我好久。

这道题要用到组合数和容斥,因为交集为k的方案并不好求,但是交集至少为k的方案数比较好求,所以考虑容斥:

设提前选出来k个数,然后求交集至少为k+x时的方案数:(2^ 2(n-k-x)-1)*C(n-k,x),交集至少为k+x时,有2(n-k-x)个集合,则有2^ 2(n-k-x)种方案,但是要减去一个都不选的一种,乘C(n-k,x),然后容斥,对于x奇减偶加;

但是因为k是提前选出来的,以上并没有考虑k的方案,所以最后答案*C(n,k);

错误示范:

1 LL ans=0;
2     for(int i=0;i<=n-k;i++)
3     {
4         LL res=(inv(2,inv(2,n-k-i))-1)*C(n-k,i)%mod;
5         if(i%2)ans-=res;
6         else   ans+=res;
7         ans%=mod;
8     }

 

ab%c!=ab%c!!!但是如果把循环顺序改一下,可以发现其实inv(2,inv(2,n-k-i))是每次平方的(inv为快速幂):

#include<iostream>
#include<cstdio>
#define mod 1000000007
#define LL long long
using namespace std;
int n,k;
LL jc[1000010];

LL inv(LL a,LL b)
{
	LL ans=1;
	while(b)
	{
		if(b&1)ans=ans*a%mod;
		a=a*a%mod;
		b=b>>1;
	}
	return ans%mod;
}
LL C(LL n,LL m)
{
	if(!m)return 1;
	return jc[n] * inv(jc[m],mod-2) %mod * inv(jc[n-m],mod-2) %mod;
}
void init()
{
	jc[0]=1;
	for(int i=1;i<=n;i++)jc[i]=jc[i-1]*i%mod;
}
signed main()
{
//	freopen("in.txt","r",stdin);

	cin>>n>>k;init();
	LL ans=0,pf=2;
	for(int i=n-k;i>=0;i--)
	{
		LL res=(pf-1)*C(n-k,i)%mod;
		pf=pf*pf%mod;
		if(i%2)ans-=res;
		else   ans+=res;
		ans=(ans%mod+mod)%mod;
	}
	ans=(ans%mod+mod)%mod;
	ans=ans*C(n,k)%mod;
	cout<<(ans%mod+mod)%mod<<endl;
}

 

波澜前,面不惊。
posted @ 2019-06-30 17:12  Al_Ca  阅读(191)  评论(0编辑  收藏  举报
ヾ(≧O≦)〃嗷~