【NOIP2016提高A组模拟9.17】小a的强迫症

Description

Input

第一行n
第二行n个数，代表每种颜色的数量

Output

答案

Sample Input

3
2 2 1

Sample Output

3

Data Constraint

$n<=100000$

Solution

假设当前做到第i种颜色，数量为a，第1到第i-1种颜色的数量和为s
那么当前已经有s个球了，为了保证合法，必须把第i中颜色的球放一个在最后面，还是a-1个球插到前面s个球中间和两边去
想到了什么？隔板问题！
简单来说，答案就是$C_{s+a-1}^{a-1}$乘起来就行了

Code

#include<cstdio>
#include<algorithm>
#include<cmath>
#define ll long long
#define fo(i,a,b) for(int i=a;i<=b;i++)
#define mo 998244353
using namespace std;
ll jc[1010000];
ll mi(ll a,ll b)
{
    if(b==0) return 1ll;
    if(b==1) return a;
    ll c=mi(a,b/2);
    if(b%2==0) return (c*c)%mo;
    else return (((c*c)%mo)*a)%mo;
}
ll C(ll m,ll n)
{
    return ((((jc[m]*mi(jc[n],mo-2))%mo)*mi(jc[m-n],mo-2))%mo)%mo;
}
int main()
{
    int n;scanf("%d",&n);
    jc[0]=1;fo(i,1,1000000) jc[i]=((ll)i*jc[i-1])%mo;
    ll ans=1,s=0;
    fo(i,1,n)
    {
        ll x;scanf("%lld",&x);
        ans=(ans*C(x+s-1,x-1))%mo;s+=x;
    }
    printf("%lld",ans);
}