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个没有的就是
∏
j
=
1
m
(
a
[
j
]
+
i
−
1
i
−
1
)
\prod_{j=1}^m {\binom{a[j]+i-1}{i-1}}
∏j=1m(i−1a[j]+i−1),等价于把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;
}