BZOJ4710 [Jsoi2011]分特产

我们可以考虑先计算出不限制每个人必须分得至少一个特产的方案数，设为f，然后利用容斥原理得出每个人至少分得一个特产的方案数，设为g

g[i]=f[i]-sigma j=1 to i-1 C(i,j)*g[j]

那么只需考虑如何求f，设f[i][j]表示前i种特产，一共j个人，不要求每人至少分一个的方案数，则f[0][j]=1，f[i][j]=f[i-1][j]*C(a[i]+j-1,j-1)，即不考虑第i种特产的方案数乘以将第i种特产任意分配的方案数

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<ctime>
#include<algorithm>
#include<iomanip>
#include<cmath>
#include<vector>
#include<queue>
#include<stack>
#include<bitset>
#include<set>
#include<map>
using namespace std;
#define MAXN 2010
#define MAXM 1010
#define INF 1000000000
#define eps 1e-8
#define MOD 1000000007
#define ll long long
ll fac[MAXN],ine[MAXN];
int n,m;
int f[MAXN][MAXN];
int a[MAXN];
ll g[MAXN];
ll C(int n,int m){
	return fac[n]*ine[m]%MOD*ine[n-m]%MOD;
}
int main(){
	int i,j;
	fac[0]=ine[0]=ine[1]=1;
	for(i=1;i<MAXN;i++){
		fac[i]=fac[i-1]*i%MOD;
	}
	for(i=2;i<MAXN;i++){
		ine[i]=(MOD-MOD/i)*ine[MOD%i]%MOD;
	}
	for(i=1;i<MAXN;i++){
		(ine[i]*=ine[i-1])%=MOD;
	}
	scanf("%d%d",&n,&m);
	for(i=1;i<=m;i++){
		scanf("%d",&a[i]);
	}
	for(j=1;j<=n;j++){
		f[0][j]=1;
	}
	for(i=1;i<=m;i++){
		for(j=1;j<=n;j++){
			f[i][j]=f[i-1][j]*C(a[i]+j-1,j-1)%MOD;
		}
	}
	for(i=1;i<=n;i++){
		g[i]=f[m][i];
		for(j=1;j<i;j++){
			(g[i]+=MOD-C(i,j)*g[j]%MOD)%=MOD;
		}
	}
	printf("%lld\n",g[n]);
	return 0;
}

/*

*/