bzoj4710 [Jsoi2011]分特产 容斥+组合数

Description

---

JYY 带队参加了若干场ACM/ICPC 比赛，带回了许多土特产，要分给实验室的同学们。
JYY 想知道，把这些特产分给N 个同学，一共有多少种不同的分法？当然，JYY 不希望任
何一个同学因为没有拿到特产而感到失落，所以每个同学都必须至少分得一个特产。
例如，JYY 带来了2 袋麻花和1 袋包子，分给A 和B 两位同学，那么共有4 种不同的
分配方法：
A：麻花，B：麻花、包子
A：麻花、麻花，B：包子
A：包子，B：麻花、麻花
A：麻花、包子，B：麻花

N, M 不超过1000，每一种特产的数量不超过1000

Output
输出一行，不同分配方案的总数。由于输出结果可能非常巨大，你只需要输出最终结果
MOD 1,000,000,007 的数值就可以了。

Solution

---

每种至少有一个，那么答案就是至少0个没有的-至少1个没有的+至少2个没有的。。。
至少i个没有的就是${\prod }_{j=1}^m\left(\genfrac{}{}{0ex}{}{a[j]+i-1}{i-1}\right)$，等价于把a[j]个相同的球放进n-i个不同的盒子里

Code

---

#include <stdio.h>
#include <string.h>
#define rep(i,st,ed) for (int i=st;i<=ed;++i)

const int MOD=1000000007;
const int N=20005;

int ny[N],fac[N];
int a[N];

int C(int n,int m) {
	return 1LL*fac[n]*ny[m]%MOD*ny[n-m]%MOD;
}

int ksm(int x,int dep) {
	int res=1;
	for (;dep;dep>>=1) {
		(dep&1)?(res=1LL*res*x%MOD):0;
		x=1LL*x*x%MOD;
	}
	return res;
}

int main(void) {
	fac[0]=1; rep(i,1,N-1) fac[i]=1LL*fac[i-1]*i%MOD;
	rep(i,0,N-1) ny[i]=ksm(fac[i],MOD-2);
	int n,m; scanf("%d%d",&n,&m);
	rep(i,1,m) scanf("%d",&a[i]);
	int ans=0;
	rep(i,0,n-1) {
		int tot=C(n,i);
		rep(j,1,m) {
			tot=1LL*tot*C(a[j]+n-i-1,n-i-1)%MOD;
		}
		if (i&1) ans=(ans-tot+MOD)%MOD;
		else ans=(ans+tot)%MOD;
	}
	printf("%d\n", ans);
	return 0;
}