[JXOI2018]排序问题

XIV.[JXOI2018]排序问题

本题好像又不算期望罢……

根据一些简单的推理，我们发现最终答案就是

$$\dfrac{(n+m)!}{\prod\limits_{i}cnt_i!}$$

其中$cnt_i$表示有多少个数是$i$。（这很简单，因为只有每个位置一一对应才能排序成功；但是值相同的数之间两两可以相互替换，故要除掉）

我们发现，对于$i\notin[l,r]$，这个$cnt_i$就已经固定了，可以直接通过哈希表/离散化预处理出来；而对于$i\in[l,r]$，$cnt_i$并不固定，而我们希望它尽量平均，故我们先用这$m$个位置尽量补，补到对于$i\in[l,r]$，$cnt_i$全部相等，然后剩下的直接平均分即可。

代码：

#include<bits/stdc++.h>
using namespace std;
const int mod=998244353;
const int N=10200000;
int T,n,m,l,r,a[200100],b[200100],c[200100],fac[10200100],inv[10200100],res;
vector<int>v;
int ksm(int x,int y){
	int z=1;
	for(;y;y>>=1,x=1ll*x*x%mod)if(y&1)z=1ll*z*x%mod;
	return z;
}
int main(){
	fac[0]=1;for(int i=1;i<=N;i++)fac[i]=1ll*fac[i-1]*i%mod;
	inv[N]=ksm(fac[N],mod-2);for(int i=N-1;i>=0;i--)inv[i]=1ll*inv[i+1]*(i+1)%mod;
	scanf("%d",&T);
	while(T--){
		scanf("%d%d%d%d",&n,&m,&l,&r),res=fac[n+m],v.clear();
		for(int i=1;i<=n;i++)scanf("%d",&a[i]),v.push_back(a[i]);
		sort(v.begin(),v.end()),v.resize(unique(v.begin(),v.end())-v.begin());
		for(int i=1;i<=n;i++)b[lower_bound(v.begin(),v.end(),a[i])-v.begin()]++;
		c[0]=r-l+1;
		for(int i=0;i<v.size();b[i++]=0)if(l<=v[i]&&v[i]<=r)c[0]--,c[b[i]]++;else res=1ll*res*inv[b[i]]%mod;
		int lim=0;
		for(lim=0;c[lim]!=r-l+1;lim++){
			if(c[lim]<=m)m-=c[lim],c[lim+1]+=c[lim],c[lim]=0;
			else{c[lim+1]+=m,c[lim]-=m,m=0;break;}
		}
		if(c[lim]==r-l+1){
			int tmp=m%c[lim],now=m/c[lim];
			res=1ll*res*ksm(inv[lim+now+1],tmp)%mod*ksm(inv[lim+now],c[lim]-tmp)%mod;
		}else for(int i=lim,j=0;j!=r-l+1;i++)res=1ll*res*ksm(inv[i],c[i])%mod,j+=c[i];
		for(int i=0;i<=n;i++)c[i]=0;
		printf("%d\n",res);
	}
	return 0;
}