[JXOI2018]排序问题

一、题目

点此看题

二、解法

0x01 50分暴力
先考虑得到一个序列计算它的期望，如果这个序列两两不同，假设长度为$n$，那么期望就是$n!$，考虑序列有重复的情况，发现相同的数的顺序不用考虑，设每个数的出现次数为$t_i$，那么期望就是$n!/t_i!$。
在考虑如果加入数字，观察上面期望的式子，发现在$[l,r]$出现数的总和一定的情况下分布越平均越好，这样就得到了一个暴力的写法，向$[l,r]$中暴力填数，保证尽量出现的数平均，然后按照上式计算答案。
0x02 二分
因为尽量平均，有一个单调性，可以考虑二分，二分平均出现的次数，然后计算达到这个次数的情况下填数的次数，把这个找到不大于$m$的最大的平均出现次数，具体操作是找到$a$数组中的$[l,r]$的数生成$b$数组，二分之后分成三部分计算，设二分的答案是$p$，第一部分就是还剩下的次数用上去，就是$fac[p+1]$，第二部分就是平均的出现次数，就是$fac[p]$，第三部分是$a$数组中出现次数大于$p$的，单独考虑。

第一次在考试中切掉紫题qwq，太激动了，不知所言。

#include <cstdio>
#include <algorithm>
using namespace std;
#define int long long
const int MAXN = 200005;
const int MAXM = 11000005;
const int MOD = 998244353;
int read()
{
    int num=0,flag=1;char c;
    while((c=getchar())<'0'||c>'9')if(c=='-')flag=-1;
    while(c>='0'&&c<='9')num=(num<<3)+(num<<1)+(c^48),c=getchar();
    return num*flag;
}
int T,n,m,k,l,r,cnt,ans,p,a[MAXN],b[MAXN];
int fac[MAXM];
void init()
{
	fac[0]=1;
	for(int i=1;i<MAXM;i++)
		fac[i]=fac[i-1]*i%MOD;
}
int qkpow(int a,int b)
{
	int res=1;
	while(b>0)
	{
		if(b&1) res=res*a%MOD;
		a=a*a%MOD;
		b>>=1;
	}
	return res;
}
int check(int x)
{
	int sum=(r-l+1-k)*x;
	for(int i=1;i<=k;i++)
		if(b[i]<x)
			sum+=x-b[i];
	return sum;
}
void solve(int l,int r)
{
	if(l>r) return ;
	int mid=(l+r)>>1;
	if(check(mid)<=m)
	{
		p=mid;
		solve(mid+1,r);
	}
	else
		solve(l,mid-1);
}
int inv(int x)
{
	return qkpow(fac[x],MOD-2);
}
signed main()
{
	//freopen("fuck.in","r",stdin);
	//freopen("AK.out","w",stdout);
	T=read();
	init();
	while(T--)
	{
		n=read();m=read();l=read();r=read();
		k=cnt=0;ans=fac[n+m];
		for(int i=1;i<=n;i++)
			a[i]=read();
		a[n+1]=0;
		sort(a+1,a+1+n);
		for(int i=1;i<=n;i++)
		{
			cnt++;
			if(a[i]!=a[i+1])
			{
				if(l<=a[i] && a[i]<=r)
					b[++k]=cnt;
				else
					ans=ans*inv(cnt)%MOD;
				cnt=0;
			}
		}
		solve(0,1e8);
		m-=check(p);
		for(int i=1;i<=k;i++)
			if(b[i]>p)
			{
				ans=ans*inv(b[i])%MOD;
				cnt++;
			}
		ans=ans*qkpow(inv(p+1),m)%MOD*qkpow(inv(p),r-l+1-m-cnt)%MOD;
		printf("%lld\n",ans);
	}
}