【BZOJ5322】[JXOI2018]排序问题（模拟）

【BZOJ5322】[JXOI2018]排序问题（模拟）

题面

BZOJ
洛谷

题解

这题就显得很呆。
显然就是每次找到\([l,r]\)中出现次数最小的那个数并且放一个。
然后随便模拟一下就好了QwQ。

#include<iostream>
#include<cstdio>
#include<algorithm>
using namespace std;
#define ll long long
#define MOD 998244353
#define TOT 12000001
#define MAX 2000200
inline int read()
{
	int x=0;bool t=false;char ch=getchar();
	while((ch<'0'||ch>'9')&&ch!='-')ch=getchar();
	if(ch=='-')t=true,ch=getchar();
	while(ch<='9'&&ch>='0')x=x*10+ch-48,ch=getchar();
	return t?-x:x;
}
int jc[TOT];
int fpow(int a,int b){int s=1;while(b){if(b&1)s=1ll*s*a%MOD;a=1ll*a*a%MOD;b>>=1;}return s;}
int n,m,l,r,a[MAX],b[MAX];
int main()
{
	jc[0]=1;for(int i=1;i<TOT;++i)jc[i]=1ll*jc[i-1]*i%MOD;
	int T=read();
	while(T--)
	{
		n=read();m=read();l=read();r=read();
		for(int i=1;i<=n;++i)a[i]=read();
		sort(&a[1],&a[n+1]);
		int f=r-l+1,ans=1,tot=0,cnt=0;
		for(int i=1,j=1;i<=n;i=j)
		{
			while(j<=n&&a[i]==a[j])++j;
			if(a[i]<l||a[i]>r)ans=1ll*ans*jc[j-i]%MOD;
			else b[++tot]=j-i,f-=1;
		}
		sort(&b[1],&b[tot+1]);
		for(int i=1,j=1;i<=tot;i=j)
		{
			while(j<=tot&&b[i]==b[j])++j;
			++cnt;a[cnt]=b[i];b[cnt]=j-i;
		}
		a[++cnt]=1e9;b[cnt]=1;
		for(int i=1,M=m;i<=cnt;++i)
		{
			int s=a[i]-a[i-1];
			ll val=1ll*s*f;
			if(val>M)
			{
				for(int j=i;j<cnt;++j)ans=1ll*ans*fpow(jc[a[j]],b[j])%MOD;
				int t=a[i-1]+M/f,num=M%f;
				ans=1ll*ans*fpow(jc[t],f-num)%MOD*fpow(jc[t+1],num)%MOD;
				break;
			}
			else M-=val,f+=b[i];
		}
		ans=1ll*jc[n+m]*fpow(ans,MOD-2)%MOD;
		printf("%d\n",ans);
	}
	return 0;
}