Description
Solution
从后往前做,很显然当前颜色做完以后就可以直接全部删掉,
直接用C公式即可。
复杂度: O(nlog(mo−2)) ;
Code
#include<cstdio>
#define fo(i,a,b) for(int i=a;i<=b;i++)
#define fod(i,a,b) for(int i=a;i>=b;i--)
using namespace std;
typedef long long LL;
const int N=100500,mo=998244353;
int read(int &n)
{
char ch=' ';int q=0,w=1;
for(;(ch!='-')&&((ch<'0')||(ch>'9'));ch=getchar());
if(ch=='-')w=-1,ch=getchar();
for(;ch>='0' && ch<='9';ch=getchar())q=q*10+ch-48;n=q*w;return n;
}
int n,m;
int a[N];
LL jc[N*5],jcn[N*5],ans;
LL ksm(LL q,LL w)
{
LL ans=1;
while(w)
{
if(w&1)ans=(ans*q)%mo;
q=(q*q)%mo;w>>=1;
}
return ans;
}
LL C(int n,int m){return jc[m]*jcn[n]%mo*jcn[m-n]%mo;}
int main()
{
int q,w;
read(n);
fo(i,1,n)m+=read(a[i]);
jc[0]=jcn[0]=1;
fo(i,1,m)jc[i]=jc[i-1]*i%mo,jcn[i]=ksm(jc[i],mo-2);
ans=1;
fod(i,n,1)ans=(ans*C(a[i]-1,m-1))%mo,m-=a[i];
printf("%lld\n",ans);
return 0;
}