码迷,mamicode.com
首页 > 其他好文 > 详细

bzoj2839 集合计数

时间:2016-05-30 15:38:42      阅读:199      评论:0      收藏:0      [点我收藏+]

标签:

2839: 集合计数

Time Limit: 10 Sec  Memory Limit: 128 MB
Submit: 243  Solved: 129
[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;




容斥原理

类似bzoj3198的思路,f[i]表示交集的个数至少是i个的方案数,最终答案ans=∑f[i]*C(i,k)*(-1)^(i-k),k≤i≤n。

于是问题转化为如何求f[i]。

加入限制了交集必须有i个数,等于将可选的集合范围缩小到了2^(n-i)个。而每一个集合都是可选可不选的,但是不能全部不选,所以f[i]=2^[2^(n-i)]-1。

可以发现随着i递减,2^[2^(n-i)]每一次都是上一次的平方,然后倒着算就好了。




#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cmath>
#include<cstring>
#include<algorithm>
#define F(i,j,n) for(int i=j;i<=n;i++)
#define D(i,j,n) for(int i=j;i>=n;i--)
#define ll long long
#define maxn 1000005
#define mod 1000000007
using namespace std;
int n,k;
ll ans,fac[maxn],inv[maxn];
inline int read()
{
	int x=0,f=1;char ch=getchar();
	while (ch<'0'||ch>'9'){if (ch=='-') f=-1;ch=getchar();}
	while (ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
	return x*f;
}
inline ll getpow(ll x,ll y)
{
	ll ret=1;
	for(;y;y>>=1,x=x*x%mod) if (y&1) ret=ret*x%mod;
	return ret;
}
int main()
{
	n=read();k=read();
	fac[0]=1;
	F(i,1,n) fac[i]=fac[i-1]*i%mod;
	inv[0]=1;inv[1]=1;
	F(i,2,n) inv[i]=inv[i-mod%i]*(mod/i+1)%mod;
	F(i,2,n) inv[i]=inv[i]*inv[i-1]%mod;
	ll x=2;
	D(i,n,k)
	{
		if (i!=n) x=x*x%mod;
		ll tmp=fac[n]*inv[n-i]%mod*inv[k]%mod*inv[i-k]%mod;
		ans=(ans+tmp*(x+mod-1)%mod*((i-k)&1?mod-1:1)%mod)%mod;
	}
	printf("%lld\n",ans);
	return 0;
}


bzoj2839 集合计数

标签:

原文地址:http://blog.csdn.net/aarongzk/article/details/51520454

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!