题目传送门
思路
一个非常神奇的思路。
先挂一个引理:Raney 引理。
这一个引理就是说:有 m m m 个数之和为 1 1 1,那么满足使得所有前缀和为正数的排列的数量为 ( m − 1 ) ! (m-1)! (m−1)!。
但是这跟原题有一点区别。原题的数之和为 0 0 0 且前缀和为非负整数,所以不能直接做。
那么我们考虑让题目变成引理的形式。如果说在数列的末尾加上一个 − 1 -1 −1,是不是就可以用引理了呢?
具体的解释可以看题解。
我们现在就可以统计答案了。由于我们在数列末尾加了一个 − 1 -1 −1,所以排列数量就变成 m ! m! m!。数列原有 m − n m-n m−n 个 − 1 -1 −1,现在加了一个,变成 m − n + 1 m-n+1 m−n+1 个,排列就有 ( m − n + 1 ) ! (m-n+1)! (m−n+1)! 个。因为多了一个 − 1 -1 −1,我们为了除去它带来的影响,要除一个 m − n + 1 m-n+1 m−n+1。答案即为 m ! m − n + 1 \dfrac{m!}{m-n+1} m−n+1m!。
代码
#include<bits/stdc++.h>
using namespace std;
#define int long long
#define endl '\n'
inline int read(){
int x=0,f=1;char c=getchar();
for(;!isdigit(c);c=getchar()) if(c=='-') f=-1;
for(;isdigit(c);c=getchar()) x=(x<<3)+(x<<1)+c-'0';
return x*f;
}
inline void write(int x){
if(x<0) putchar('-'),x=-x;
if(x>9) write(x/10);
putchar(x%10+'0');
}
const int maxn=45,mod=998244353;
int n,m,ans=1;
signed main(){
n=read();
for(int i=1;i<=n;i++) m+=read();
for(int i=1;i<=m;i++) if(i!=m-n+1) ans*=i,ans%=mod;
cout<<ans%mod<<endl;
return 0;
}