Bzoj4710 分特产(容斥原理+组合数)

题面

Bzoj

题解

考虑容斥原理,所有人都有特产的方案数等于:

至少零个人没有特产$-$至少一个人没有特产$+$至少两个人有特产$-…$

接着考虑其中一种情况怎么求(假设现在至少有$i$个人没有特产):

对于每种特产,我们分开考虑,假设当前特产有$a[j]$个,则我们可以看作是将$a[j]$个相同的球放入$n-i$个相同的盒子中,允许出现空盒(因为之前说的是至少),利用插板法,方案数为:$C_{n-i+a[j]-1}^{n-i-1}$

最后当前情况的贡献绝对值就是,$C[n][i]$乘上每种特产的贡献之积,$C[n][i]$表示使得$n$个同学中的$i$个没有特产。

#include <cstdio>
#include <cstring>
#include <algorithm>
using std::min; using std::max;
using std::swap; using std::sort;
typedef long long ll;

template<typename T>
void read(T &x) {
    int flag = 1; x = 0; char ch = getchar();
    while(ch < '0' || ch > '9') { if(ch == '-') flag = -flag; ch = getchar(); }
    while(ch >= '0' && ch <= '9') x = x * 10 + ch - '0', ch = getchar(); x *= flag;
}

const int N = 2e3 + 10, P = 1e9 + 7;
int n, m, a[N], c[N][N], ret;

int main () {
    int lim = N - 10; c[0][0] = 1;
    for(int i = 1; i <= lim; ++i) {
        c[i][0] = c[i][i] = 1;
        for(int j = 1; j < i; ++j)
            c[i][j] = (c[i - 1][j] + c[i - 1][j - 1]) % P;
    }
    read(n), read(m);
    for(int i = 1; i <= m; ++i) read(a[i]);
    for(int i = 0, gx = 1; i <= n; ++i) {
        int dq = 1;
        for(int j = 1; j <= m; ++j)
            dq = 1ll * dq * c[n + a[j] - i - 1][n - i - 1] % P;
        if(gx > 0) ret = (ret + 1ll * c[n][i] * dq % P) % P;
        else ret = (ret + P - 1ll * c[n][i] * dq % P) % P;
        gx = -gx;
    } printf("%d\n", ret);
    return 0;
}