小a的强迫症 组合数学

模拟赛   T1

看题意可以很清楚的知道，这是一道组合数学的题（废话）。考虑每种颜色的珠子放的方法数，最后乘法原理把它们相乘即可。

（别忘了取模！！！）

因为第i种珠子的最后一颗要在第i+1种的最后一颗后，所以从最后一个空位、最后一种珠子开始枚举。sum[i]表示前i种颜色的珠子总和。

刚开始，最后的空位只能放最后一种珠子，然后把剩下的这种颜色的珠子往前面sum[i]-1个空里放，有C(sum[i]-1,num[i]-1)种。

同理，每往前枚举一种颜色的珠子此时这种珠子之后的其他颜色的珠子已经放完了，所以当前的最后一个空位一定要放这种珠子，然后剩下的放在前面sum[i]-1个空里，依然是C(sum[i]-1,num[i]-1)种。

所以公式就是：$$\Pi_{i=1}^{n}C^{num_{i}-1}_{sum_{i}-1}$$

因为要取模所以还要用逆元，用费马小定理处理一下就好。

#include<iostream>
#include<cstdio>
#include<cmath>
#include<cstring>
#include<algorithm>
#define ll long long
#define MOD 998244353
using namespace std;
ll n,num[100005],jc[500005];
ll ksm(ll x,ll y)
{
	ll ans=1;
	while(y)
	{
		if(y&1) ans=(ans*x)%MOD;
		x=(x*x)%MOD;
		y>>=1;
	}
	return ans;
}
ll inv(ll x)
{
	return ksm(x,MOD-2)%MOD;
}
ll c(ll n,ll m)
{
	if(m>n) return 0;
	return (((jc[n]*inv(jc[m]))%MOD)*inv(jc[n-m]))%MOD;
}
int main()
{
	freopen("qiang.in","r",stdin);
	freopen("qiang.out","w",stdout); 
	scanf("%lld",&n);
	ll s=0;
	for(int i=1;i<=n;i++) 
		scanf("%lld",&num[i]),s+=num[i];
	jc[0]=1;
	for(ll i=1;i<=s;i++) jc[i]=(jc[i-1]*i)%MOD;
	ll ans=1;
	for(ll i=n;i>=1;i--)
	{
		ans=(ans*c(s-1,num[i]-1))%MOD;
		s-=num[i];
	}
	printf("%lld",ans);
	return 0;
}