JYY 带队参加了若干场ACM/ICPC 比赛,带回了许多土特产,要分给实验室的同学们。
JYY 想知道,把这些特产分给N 个同学,一共有多少种不同的分法?当然,JYY 不希望任何一个同学因为没有拿到特产而感到失落,所以每个同学都必须至少分得一个特产。
例如,JYY 带来了2 袋麻花和1 袋包子,分给A 和B 两位同学,那么共有4 种不同的分配方法:
A:麻花,B:麻花、包子
A:麻花、麻花,B:包子
A:包子,B:麻花、麻花
A:麻花、包子,B:麻花
输入
输入数据第一行是同学的数量N 和特产的数量M。
第二行包含M 个整数,表示每一种特产的数量。
N, M 不超过1000,每一种特产的数量不超过1000
输出
输出一行,不同分配方案的总数。由于输出结果可能非常巨大,你只需要输出最终结果 MOD 1,000,000,007 的数值就可以了。
样例输入
5 4
1 3 3 5
样例输出
384835
由于“每个同学都必须至少分得一个特产”这个限制比较难处理,所以我们可以考虑容斥,
用没有限制-至少1个人没分到+至少2个人没分到-… 得到答案。
考虑如果i个人没分到该怎么处理:n个人选出i个不分,方案数为C(n,i);
对于每种特产设有w[i]个,分给(n−i)个同学,允许有人拿不到,方案数为
c(w[i]+n-i-1,n-i-1)
故最终答案为∑(-1)^i * C(n,i) *∑C(w[j]+n-i-1,n-i-1) (0<=i<=n-1,1<=j<=m)
手动算一下
3 2
3 4
则没有限制的有c(3,0)*c(5,2)*C(6,2)=150
至少有一个人没拿到C(3,1)*C(4,1)*C(5,1)=60
至少有二个人没拿到C(3,2)*1*1=3
ans=150-60+3=93
#include
#define N 2010
using namespace std;
typedef long long ll;
const ll mod = 1000000007;
ll c[N][N];
int w[N];
int main()
{
int n , m , i , j;
ll ans = 0 , tmp;
for(i = 0 ; i <= 2000 ; i ++ )
{
c[i][0] = 1;
for(j = 1 ; j <= i ; j ++ )
c[i][j] = (c[i - 1][j - 1] + c[i - 1][j]) % mod;
}
scanf("%d%d" , &n , &m);
for(i = 1 ; i <= m ; i ++ ) //M个特产
scanf("%d" , &w[i]);
for(i = 0 ; i < n ; i ++ )
{
tmp = c[n][i];
// cout<<" i is "<<i<<endl;
for(j = 1 ; j <= m ; j ++ )
{
tmp = tmp * c[ w[j] + n - i - 1] [w[j]] % mod;
// cout<<" m iss "<<c[ w[j] + n - i - 1] [w[j]]<<endl;
}
// cout<<i<<" tmp is "<<tmp<<endl;
if(i & 1)
ans = (ans - tmp + mod) % mod;
else
ans = (ans + tmp) % mod;
// cout<<i<<" ans is "<<ans<<endl;
}
printf("%lld\n" , ans);
return 0;
}
手机扫一扫
移动阅读更方便
你可能感兴趣的文章