CodeForces 451E Devu and Flowers（容斥原理+组合数学+状态压缩）

最新推荐文章于 2020-12-03 22:58:42 发布

godspeedkaka

最新推荐文章于 2020-12-03 22:58:42 发布

阅读量993

点赞数

分类专栏：程序设计竞赛文章标签：容斥原理数论状态压缩

本文链接：https://blog.csdn.net/u014664226/article/details/51485557

版权

程序设计竞赛专栏收录该内容

360 篇文章 0 订阅

订阅专栏

题意：给出n堆花，对于第j堆，有f[j]朵花，每堆花的颜色不同，现在要从中选出s朵，求方案数。

思路：这道题如果值借用背包做的话复杂度爆炸

现在假如没有每堆花的容量限制，那么我们可以用挡板法求出方案数，所以说我们可以先用挡板法先求出总方案数再减去其中不合法的方案数

所以我们可以状态压缩枚举不合法的堆，然后用容斥原理来做。

#include <bits/stdc++.h>
#define eps 1e-6
#define LL long long
#define pii pair<int, int>
#define pb push_back
#define mp make_pair
//#pragma comment(linker, "/STACK:1024000000,1024000000")
using namespace std;

//const int MAXN = 5000000 + 5;
const int MOD = 1e9+7;
//const int INF = 0x3f3f3f3f;
int n;
LL s, f[30];
LL pow_mod(int a, int b) {
	if (!b) return 1;
	LL ans = pow_mod(a, b/2);
	ans = ans * ans % MOD;
	if (b&1) 
		ans = ans * a % MOD;
	return ans;
}
LL get_c(LL a, LL b) {
	LL tmp1 = 1, tmp2 = 1;
	for (int i = 1; i <= b; i++) {
		tmp1 = (a-i+1)%MOD * tmp1 % MOD;
		tmp2 = tmp2 * i % MOD;
	}
	return tmp1 * pow_mod(tmp2, MOD-2) % MOD;
}
int main()
{
	//freopen("input.txt", "r", stdin);
	scanf("%d%I64d", &n, &s);
	for (int i = 0; i < n; i++)
		scanf("%I64d", &f[i]);
	int maxS = 1 << n;
	LL ans = 0;
	for (int i = 0; i < maxS; i++) {
		LL sumv = 0;
		int flag = 1;
		for (int j = 0; j < n; j++) if ((1<<j) & i) {
			sumv += f[j] + 1;
			flag *= -1;
		}
		if (sumv > s) continue;
		ans = (ans+flag*get_c(s-sumv+n-1, n-1)) % MOD;
	}
	printf("%I64d", (ans+MOD)%MOD);
	return 0;
}