2021-11-01

题目描述

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;
}