bzoj 4710: [Jsoi2011]分特产 （容斥原理+DP）

题目描述

传送门

题目大意：n个同学，m种特产，每个特产有ai个。求每个同学至少分得一个特产的方案数。

题解

根据容斥原理：
没有空盒子的方案数=至少0个空盒子的方案数-至少1个空盒子的方案数+至少2个空盒子的方案数…….
那么我们可以对于每个特产分开考虑，相当于是n个球放到m个盒子中,可以有空盒子，方案数可以同插板法计算，答案是 C(n+m−1,m−1) .那么如何考虑至少i个空盒子的贡献呢？我们直接选出i个盒子(方案数为 C(n,i) )，强制他们为空，剩下的利用插板法即可。

代码

#include
#include
#include
#include
#include
#define N 2000
#define p 1000000007
#define LL long long 
using namespace std;
int n,m,a[N+3],c[N+3][N+3];
int main()
{
    freopen("a.in","r",stdin);
    scanf("%d%d",&n,&m);
    for (int i=0;i<=N;i++) c[i][0]=1;
    for (int i=1;i<=N;i++)
     for (int j=1;j<=i;j++) c[i][j]=(c[i-1][j-1]+c[i-1][j])%p;
    for (int i=1;i<=m;i++) scanf("%d",&a[i]);
    int t; LL sum=0;
    for (int i=0;i<=n;i++) {
        if (i&1) t=-1;
        else t=1;
        LL ans=1;
        for (int j=1;j<=m;j++) ans=ans*c[n+a[j]-i-1][n-i-1]%p;
        sum=(sum+t*c[n][i]%p*ans%p)%p;
    }
    printf("%lld\n",(sum%p+p)%p);
}