主要思路
这道题需要多重集的知识,如果没有学习请看这篇博客的多重集部分。
很明显,题意要求我们求出多重集的组合数,且为“增强版组合数”。所以,我们根据公式,使用二进制分解的方法来求出即可。这是一道比较裸的多重集组合数问题。
代码
// CF451E.cpp
#include <bits/stdc++.h>
#define ll long long
using namespace std;
const int mod = 1e9 + 7, MAX_N = 25;
ll arr[MAX_N], n, inv[MAX_N], m;
ll quick_power(ll bas, ll tim)
{
ll ans = 1;
while (tim)
{
if (tim & 1)
ans = ans * bas % mod;
bas = bas * bas % mod;
tim >>= 1;
}
return ans;
}
ll C(ll a, ll b)
{
if (a < 0 || b < 0 || a < b)
return 0;
a %= mod;
if (a == 0 || b == 0)
return 1;
ll ans = 1;
for (int i = 0; i < b; i++)
ans = ans * (a - i) % mod;
for (int i = 1; i <= b; i++)
ans = ans * inv[i] % mod;
return ans;
}
int main()
{
scanf("%lld%lld", &n, &m);
for (int i = 1; i <= n; i++)
scanf("%lld", &arr[i]);
inv[0] = 1;
for (int i = 1; i < MAX_N; i++)
inv[i] = quick_power(i, mod - 2);
ll ans = 0;
for (int stat = 0; stat < (1 << n); stat++)
if (stat == 0)
ans = (ans + C(n + m - 1, n - 1)) % mod;
else
{
ll t = n + m, p = 0;
for (int i = 0; i < n; i++)
if ((stat >> i) & 1)
p++, t -= arr[i + 1];
t -= (p + 1);
if (p & 1)
ans = (ans - C(t, n - 1)) % mod;
else
ans = (ans + C(t, n - 1)) % mod;
}
ans = (ans + mod) % mod;
printf("%lld", ans);
return 0;
}
source:Page 172,《算法竞赛进阶指南》李煜东著