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

时间:2022-06-12 18:59:44

题意:给出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;
}