cf 553A 排列组合

题意 k种颜色,第i种颜色Ci个balls

求第i种颜色的最后一颗球必须在第i+1种颜色的最后一颗球之前

数学排列组合公式

C(n,m)=C(n-1,m)+C(n-1,m-1)

思路:

总的球数为sum,球数为a[i]

从当前最后一种颜色思考,这种颜色的最后一颗球必须放在最后一个位置

将这种颜色放好后还剩sum-1颗球,这种颜色的球还剩a[i]-1

那么对于这a[i]-1颗球,就放在这sum-1中,这些球的位置在这些位置中可以自由放置,那么就有C(sum-1,a[i]-1)种放法

剩下了sum-a[i]个位置,对于这些位置我们可以看作是连续的sum-a[i]个位置

那么之后就是对a[i-1]的球在sum-a[i]个位置中放置,重复上面步骤,每次的放法之积就是结果

代码如下

#include <cstdio>
#include <algorithm>
#include <iostream>
#define MOD 1000000007
#define MAX 1005
int ans[MAX][MAX];
int mp[MAX];
using namespace std;
int main(void){
for(int i = 0;i<MAX;i++)ans[i][0]=1;//初始化
for(int i = 1;i<MAX;i++)
for(int j = 1;j <= i;j++){
ans[i][j]=(ans[i-1][j]+ans[i-1][j-1])%MOD;
}
int sum = 0;
long long cnt = 1;
int k;cin >> k;
for(int i = 1;i<=k;i++)cin >> mp[i],sum += mp[i];
for(int i = k;i>=1;i--){
cnt = (cnt*ans[sum-1][mp[i]-1])%MOD;
sum-= mp[i];
}
cout << cnt << endl;
}


你可能感兴趣的:(c)