分析
感觉这道题的计数方法好厉害。。
一个直观的思路是,把题目转化为求至少有(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;
}