51nod-1453 抽彩球

原题链接

1453 抽彩球

题目来源：  CodeForces

基准时间限制：1 秒 空间限制：131072 KB 分值: 40  难度：4级算法题

 收藏

 关注

一个袋子中有n个彩球，他们用k种不同的颜色染色。颜色被从1到k编号。同一种颜色的球看成是一样的。现在从袋中一个一个的拿出球来，直到拿完所有的球。对于所有颜色为i (1<=i<=k-1)的球，他的最后一个球总是在编号比他大的球拿完之前拿完，问这样情况有多少种。

样例解释：这个样例中有2个1号颜色的球，2个2号颜色的球，1个3号颜色的球。三种方案是：
1 2 1 2 3
1 1 2 2 3
2 1 1 2 3

Input

单组测试数据。
第一行给出一个整数k(1 ≤ k ≤ 1000)，表示球的种类。
接下来k行，每行一个整数ci,表示第i种颜色的球有ci个(1 ≤ ci ≤ 1000)。
球的总数目不超过1000。

Output

输出总数对1,000,000,007的模即可。

Input示例

3
2
2
1

Output示例

3

dp[i][j]表示在第j个位置放i颜色的最后一个球时有多少种情况

#include <cstdio>
#include <iostream>
#define maxn 1005 
#define MOD 1000000007
typedef long long ll;
using namespace std;

ll dp[maxn][maxn];
ll p1[maxn], p2[maxn];
int num[maxn];
ll solve(ll d){
	ll ans = 1;
	ll e = MOD - 2;
	while(e){
		if(e&1)
		 (ans *= d) %= MOD;
		(d *= d) %= MOD;
		e >>= 1;
	}
	return ans;
}
int main(){
	
//	freopen("in.txt", "r", stdin);
	int n, sum = 0;
	scanf("%d", &n);
	for(int i = 1; i <= n; i++){
	 scanf("%d", num+i);
	 sum += num[i];
    }
	p1[0] = 1;
	p2[0] = solve(p1[0]);
	for(int i = 1; i <= sum; i++){
		(p1[i] = i * p1[i-1]) %= MOD;
		p2[i] = solve(p1[i]);
	}
	int m = 0;
	for(int i = 0; i <= sum; i++)
	 dp[0][i] = 1;
	for(int i = 1; i <= n; i++){
		int l = m + num[i], r = sum - (n - i);
		for(int j = l; j <= r; j++){
			int k = j - m - 1;
			dp[i][j] = dp[i-1][j-1] * p1[k] % MOD * p2[k-num[i]+1] % MOD * p2[num[i]-1] % MOD;
			(dp[i][j] += dp[i][j-1]) %= MOD;
		} 
		m += num[i];
	}
	printf("%I64d\n", dp[n][sum]);
	return 0;
}