题目描述
Zztrans 决定和他的同学们去拍一张班级合照,为了让每个人都能在照片里出镜,他们约定按如下方式站队进行拍照:排成人数相同的两排,每排从左向右身高都不递减,且第二排同学的身高不低于第一排对应位置同学的身高。
Zztrans 擅长排序,他得到了每个同学身高从低到高排序后的排名(身高相同的同学排名相同),这样他就不用一个个统计每个同学的身高了。但是他并不擅长数数,希望你教教他一共有多少种排队的方案。
由于答案可能很大,你只需要输出答案对 998244353取模后的结果。
输入描述
第一行有一个整数 n(2≤n≤5000),表示 Zztrans 和他同学们的总人数,保证 n 为偶数。
第二行有 n 个整数 a1,a2…,an ,(1<=ai<=n)中间以空格分隔,分别表示每个同学的身高排名。
输出描述
在一行输出一个整数,表示 Zztrans 和他的同学们按约定可能的排队方案数对 998244353 取模后的结果。
题目分析
二维DP,f[i][k] ,i表示当用上1~i的身高时,第二维k是第一排同学的数量当前为k。当将升高为i的同学有j位排在第一排时,剩余身高为i的同学自然排到第二排。当用完身高为i的同学时,要满足第二排的同学数量小于等于第一排的。时间复杂度O(n^2)。
代码
#include<iostream>
using namespace std;
typedef long long LL;
const int mod = 998244353;
const int N = 5e3+5;
LL f[N][N];
LL fan[N];
void init(){
fan[0]=1;
for(int i=1;i<N;++i)fan[i]=fan[i-1]*i%mod;
}
int a[N],cnt[N],q[N];
int main(){
int n;
init();
scanf("%d",&n);
for(int i=1;i<=n;++i){
scanf("%d",a+i);
cnt[a[i]]++;
}
for(int i=1;i<N;++i){
q[i]=q[i-1]+cnt[i];
}
f[0][0]=1;
for(int i=1;i<N;++i){
for(int j=0;j<=cnt[i];++j){
for(int k=q[i-1];k>=0;k--){
if(q[i]-k-j>=k+j){
f[i][k+j]=(f[i][k+j]+f[i-1][k]*fan[cnt[i]]%mod)%mod;
}
}
}
}
printf("%lld\n",f[N-1][n/2]);
return 0;
}