51nod 1453 抽彩球

34 篇文章 0 订阅
15 篇文章 0 订阅

对于每种颜色,为了符合题意可以先将最后的球单独考虑。

对于第1种颜色,除去最后的球,有 c[1]-1 个球。在为空的序列上随机放置,有 C(c[i]-1,c[i]-1)=1 种放法,然后将被单独考虑的球放在末尾。

对于第2种颜色, 除去最后的球,有 c[2]-1 个球。在已有c[1]个球的序列上随机放置,有 C(c[1]+c[2]-1,c[2]-1) 种方法,然后将被单独考虑的球放在末尾。

……

对于第 i 种颜色,除去最后的求,有 c[i]-1 个球。在已有 sum(c[],1,i-1) 个球的序列上随机放置,有 C(sum(c[],1,i-1)+c[i]-1,c[i]-1) 种放法,然后将被单独考虑的球放在末尾。

……

将以上的组合数相乘,即是结果。

求组合数时,用到了乘法逆元,因为1e9+7是素数,可以用费马小定理求解。

#include<bits/stdc++.h>
using namespace std;

const long long mod=1e9+7;

long long c[1010],fact[1010];

long long powmod(long long x,long long p)
{
	long long ret=1;
	while(p)
	{
		if(p&1)
			ret=ret*x%mod;
		x=x*x%mod;
		p>>=1;
	}
	return ret;
}

long long inv(long long x)
{
	return powmod(x,mod-2);
}

long long C(long long n,long long m)
{
	if(m>n)
		return 0; 
	return fact[n]*inv(fact[m])%mod*inv(fact[n-m])%mod;
}

int main()
{
	long long k,i,chosum,ans;
	fact[0]=1;
	for(i=1;i<=1010;i++)
	{
		fact[i]=fact[i-1]*i%mod;
	}
	while(cin>>k)
	{
		for(i=1;i<=k;i++)
			scanf("%lld",&c[i]);
		ans=1;
		chosum=0;
		for(i=1;i<=k;i++)
		{
			ans=ans*C(chosum+c[i]-1,c[i]-1)%mod;
			chosum+=c[i];
		}
		printf("%lld\n",ans);
	}
}


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值