luogu P2791 幼儿园篮球题

背景：

律师函警告。
另外$\text{yyb}$太巨了，解锁了新姿势。

题目传送门：

https://www.luogu.org/problem/P2791

题意：

$N$个球，其中$M$个没气的；每一次$\text{cxk’s fans}$从中选出$n$个球，其中$m$个没气的；而$\text{cxk}$每次会从这$n$个球中选出$k$个球，其中每一次没气的球的个数$x$会产生$x^k$的贡献，求最后的总期望。

思路：

考虑$ans$的表示，$i$枚举$\text{cxk}$选出的没气的球，很容易有下面的式子，不解释了，$\text{cxk}$水平。
$$ans=\frac{C_M^m{C}_{N-M}^{n-m}{\sum }_{i=0}^k{C}_m^i{C}_{n-m}^{k-i}{i}^L}{C_M^m{C}_{N-M}^{n-m}}$$

我们知道$i^L={\sum }_{j=0}^L{S}_{L,j}{C}_i^{j}j!$，带入可得：
$$ans=\sum _{i=0}^k{C}_m^i{C}_{n-m}^{k-i}\sum _{j=0}^L{S}_{L,j}{C}_i^{j}j!$$

$$ans=\sum _{i=0}^k{C}_m^i{C}_{n-m}^{k-i}\sum _{j=0}^L{S}_{L,j}{C}_i^{j}j!j!$$

$$ans=\sum _{j=0}^L{S}_{L,j}j!\sum _{i=0}^k{C}_m^i{C}_{n-m}^{k-i}{C}_i^j$$

搞到这里不会了啊，式子化简不出来，膜题解吧。
$$ans=\sum _{j=0}^L{S}_{L,j}j!\sum _{i=0}^k\frac{m!}{i!(m-i)!}{C}_{n-m}^{k-i}\frac{i!}{j!(i-j)!}$$

$$ans=\sum _{j=0}^L{S}_{L,j}j!\sum _{i=0}^k\frac{m!}{(m-i)!}\frac{1}{j!(i-j)!}{C}_{n-m}^{k-i}$$

$$ans=\sum _{j=0}^L{S}_{L,j}j!\sum _{i=0}^k\frac{m!}{(m-i)!(m-j)!}\frac{(m-j)!}{j!(i-j)!}{C}_{n-m}^{k-i}$$

$$ans=\sum _{j=0}^L{S}_{L,j}j!\sum _{i=0}^k\frac{m!}{j!(m-j)!}\frac{(m-j)!}{(m-i)!(i-j)!}{C}_{n-m}^{k-i}$$

$$ans=\sum _{j=0}^L{S}_{L,j}j!\sum _{i=0}^k{C}_m^j{C}_{m-j}^{i-j}{C}_{n-m}^{k-i}$$

$$ans=\sum _{j=0}^L{S}_{L,j}\frac{m!}{(m-j)!}\sum _{i=0}^k{C}_{m-j}^{i-j}{C}_{n-m}^{k-i}$$

考虑范德蒙恒等式${\sum }_{i=0}^k{C}_{m-j}^{i-j}{C}_{n-m}^{k-i}=C_{n-j}^{k-j}$，有：
$$ans=\sum _{j=0}^L{S}_{L,j}\frac{m!}{(m-j)!}{C}_{n-j}^{k-j}$$

$$ans=\sum _{j=0}^L{S}_{L,j}\frac{m!}{(m-j)!}\frac{(n-j)!}{(k-j)!(n-k)!}$$

$$ans=\frac{m!}{(n-k)!}\sum _{j=0}^L{S}_{L,j}\frac{1}{(m-j)!}\frac{(n-j)!}{(k-j)!}$$
预处理第二类斯特林数，就可以$\Theta (L)$出解了。
总时间复杂度：$\Theta (n+L\log L+SL)$，挺大的。
卡常警告。
$[1]$：$i^L$可以用线性筛处理，避免快速幂的$\text{log}$；
$[2]$：不要用逆元的前缀积来算阶乘的逆元，选择先算出最大的阶乘的逆元，每一次再乘$i+1$推过去，快了好多。

代码：

#include<cstdio>
#include<cstring>
#include<algorithm>
#define LL long long
#define I inline
#define R register
#define mod 998244353
using namespace std;
	int r[600010],f[600010],g[600010],w[600010],prime[200010],Pow[200010],fac[20000010],Inv[20000010];
	int S,L,n,m,k,limit,l,ans;
I int ksm(int x,int k)
{
	int tot=1;
	for(;k;k>>=1)
	{
		if(k&1) tot=(LL)tot*x%mod;
		x=(LL)x*x%mod;
	}
	return tot;
}
I void NTT(int *now,int limit,int n,int op)
{
	for(R int i=0;i<limit;++i)
		if(i<r[i]) std::swap(now[i],now[r[i]]);
	for(R int mid=1;mid<limit;mid<<=1)
	{
		int wn=ksm(op==1?3:332748118,(mod-1)/(mid<<1));
		w[0]=1;
		for(R int k=1;k<mid;k++)
			w[k]=(LL)w[k-1]*wn%mod;
		for(R int j=0;j<limit;j+=(mid<<1))
			for(R int k=0;k<mid;++k)
			{
				int x=now[j+k],y=(LL)w[k]*now[j+k+mid]%mod;
				now[j+k]=(x+y)%mod;
				now[j+k+mid]=(x-y+mod)%mod;
			}
	}
	if(op==1) return;
	int INV=ksm(limit,mod-2);
	for(R int i=0;i<n;++i)
		f[i]=(LL)f[i]*INV%mod;
}
I void Init()
{
	int ma=std::max(n,L);
	fac[0]=fac[1]=Inv[0]=Inv[1]=1;
	for(R int i=2;i<=ma;++i)
		fac[i]=(LL)fac[i-1]*i%mod;
	Inv[ma]=ksm(fac[ma],mod-2);
	for(R int i=ma-1;i;i--)
		Inv[i]=(LL)Inv[i+1]*(i+1)%mod;
	
	int t=0;
	Pow[1]=1;
	for(R int i=2;i<=L;++i)
	{
		if(!Pow[i]) Pow[i]=ksm(i,L),prime[++t]=i;
		for(R int j=1;j<=t&&i*prime[j]<=L;++j)
		{
			Pow[i*prime[j]]=(LL)Pow[i]*Pow[prime[j]]%mod;
			if(!(i%prime[j])) break;
		}
	}
	
	for(R int i=0;i<=L;++i)
	{
		f[i]=(i&1)?mod-Inv[i]:Inv[i];
		g[i]=(LL)Pow[i]*Inv[i]%mod;
	}
	limit=1,l=0;
	++L;
	while(limit<(L<<1))
		limit<<=1,++l;
	for(R int i=1;i<limit;++i)
		r[i]=((r[i>>1]>>1)|((i&1)<<(l-1)));
	NTT(f,limit,L,1),NTT(g,limit,L,1);
	for(R int i=0;i<limit;++i)
		f[i]=(LL)f[i]*g[i]%mod;
	NTT(f,limit,L,-1);
	--L;
}
I int read()
{
	int x=0,f=1;
	char ch=getchar();
	for(;ch<'0'||ch>'9';ch=getchar());
	for(;ch>='0'&&ch<='9';x=(x<<3)+(x<<1)+(ch^48),ch=getchar());
	return x*f;
}
void print(int x)
{
	if(x>9) print(x/10);
	putchar(x%10+'0');
}
int main()
{
	int mi;
	n=read(),m=read(),S=read(),L=read();
	Init();
	for(R int j=1;j<=S;++j)
	{
		ans=0;
		n=read(),m=read(),k=read();
		mi=std::min(std::min(k,m),L);
		for(R int i=1;i<=mi;++i)
			ans=(ans+(LL)f[i]*Inv[m-i]%mod*fac[n-i]%mod*Inv[k-i]%mod)%mod;
		print((LL)fac[m]*fac[k]%mod*Inv[n]%mod*ans%mod),puts("");
	}
}