[LOJ3119][CTS2019|CTSC2019]随机立方体：组合数学+二项式反演

分析

感觉这道题的计数方法好厉害。。

一个直观的思路是，把题目转化为求至少有(k)个极大的数的概率。

考虑这样一个事实，如果钦定((1,1,1),(2,2,2),...,(k,k,k))是那(k)个极大值的位置，并且(val(1,1,1) < val(2,2,2) < ... < val(k,k,k))。我们考虑依次确定这些值，显然(val(1,1,1))的值是和它至少有一维相同的(n imes m imes l - (n-1) imes (m-1) imes (l-1))个位置中最大的一个，(val(2,2,2))的值是和它至少有一维相同的所有位置并上((1,1,1))限制到的所有位置中最大的一个，即(n imes m imes l - (n-2) imes (m-2) imes (l-2))个位置中最大的一个。

以此类推，(val(i,i,i))的值是(n imes m imes l - (n-i) imes (m-i) imes (l-i))个位置中最大的一个。我们记(cnt(i) = n imes m imes l - (n-i) imes (m-i) imes (l-i))，进而可以得到((i,i,i))是极大值的概率是(P(i) = frac{1}{cnt(i)})。

现在我们所需要的就是计算(P(i))的前缀和，这个可以通过线性处理逆元的技巧完成，然后二项式反演，式子如下，其中(A_n^m)表示排列数：

[ans = sum_{i=k}^{n}(-1)^{i-k}inom{i}{k}A_n^iA_m^iA_l^iprod_{j=1}^{i}P(j) ]

关于概率为什么能二项式反演？

可以这样理解：概率再乘上个阶乘就是方案数了。

yyb聚聚的题解

戳这里

分析的方式不太一样，不过本质和最后得到的结果是相同的。

代码

#include <bits/stdc++.h>

#define rin(i,a,b) for(int i=(a);i<=(b);++i)
#define irin(i,a,b) for(int i=(a);i>=(b);--i)
#define trav(i,a) for(int i=head[a];i;i=e[i].nxt)
#define Size(a) (int) a.size()
#define pb push_back
#define mkpr std::make_pair
#define fi first
#define se second
#define lowbit(a) ((a)&(-(a)))
typedef long long LL;

using std::cerr;
using std::endl;

inline int read(){
	int x=0,f=1;char ch=getchar();
	while(!isdigit(ch)){if(ch=='-')f=-1;ch=getchar();}
	while(isdigit(ch)){x=x*10+ch-'0';ch=getchar();}
	return x*f;
}

const int MAXN=5000005;
const int MOD=998244353;

int n,m,l,k;
int fac[MAXN],invf[MAXN];
int cnt[MAXN],fix[MAXN];

inline int qpow(int x,int y){
	int ret=1,tt=x%MOD;
	while(y){
		if(y&1)ret=1ll*ret*tt%MOD;
		tt=1ll*tt*tt%MOD;
		y>>=1;
	}
	return ret;
}

inline int C(int n,int m){
	if(n<0||m<0||n<m)return 0;
	return 1ll*fac[n]*invf[n-m]%MOD*invf[m]%MOD;
}

inline int A(int n,int m){
	if(n<0||m<0||n<m)return 0;
	return 1ll*fac[n]*invf[n-m]%MOD;
}

void init(int n){
	fac[0]=1;
	rin(i,1,n)fac[i]=1ll*fac[i-1]*i%MOD;
	invf[n]=qpow(fac[n],MOD-2);
	irin(i,n-1,0)invf[i]=1ll*invf[i+1]*(i+1)%MOD;
}

int main(){
	init(5000000);
	int T=read();
	while(T--){
		int inp[4];
		rin(i,1,3)inp[i]=read();
		std::sort(inp+1,inp+4);
		n=inp[1],m=inp[2],l=inp[3];
		k=read();
		int tot=1;
		rin(i,1,n){
			cnt[i]=(1ll*n*m%MOD*l%MOD-1ll*(n-i)*(m-i)%MOD*(l-i)%MOD+MOD)%MOD;
			tot=1ll*tot*cnt[i]%MOD;
		}
		fix[n]=qpow(tot,MOD-2);
		irin(i,n-1,1)fix[i]=1ll*fix[i+1]*cnt[i+1]%MOD;
		int ans=0,sgn=MOD-1;
		rin(i,k,n){
			sgn=MOD-sgn;
			ans=(ans+1ll*sgn*C(i,k)%MOD*A(n,i)%MOD*A(m,i)%MOD*A(l,i)%MOD*fix[i])%MOD;
		}
		printf("%d
",ans);
	}
	return 0;
}