【CTS2019】隨機立方體（容斥）

【CTS2019】隨機立方體（容斥）

題面

LOJ
洛谷

題解

作這道題目的時候不難想到容斥的方面。
那麼咱們考慮怎麼計算至少有\(k\)個極大值的方案數。
咱們首先能夠把\(k\)個極大值的位置給肯定出來，方案數是\(\displaystyle {n\choose k}{m\choose k}{l\choose k}(k!)^3\)，乘上\(k!\)是爲了肯定之間的順序關係，即咱們先肯定\(xyz\)三維，而後把這三維要一一對應到點才行。假設這個值是\(w[k]\)。
剩下要填的是兩個部分，一個是剩下的\((n-k)(m-k)(l-k)\)個沒有什麼影響的位置，以及和極大值有至少一維座標相交的點。
因此方案數大概能夠寫成\({nml\choose nml-(n-k)(m-k)(l-k)}w[k]((n-k)(m-k)(l-k))!*h[k]\)的樣子。
其中\(h[k]\)是分配極大值所在的\(k*k*k\)的這個小立方體上的數字的方案數。
爲了方便\(V=nml,v[k]=V-(n-k)(m-k)(l-k)\)。
因此咱們只須要考慮怎麼分配這個挖掉\(k\)層的合法方案數。
首先每次肯定掉一個極大值以後，咱們就能夠把它所在的這三個面直接丟掉，變成小一圈的一個立方體，而對於極大值而言，由於它要比全部同層的數都要大，因此咱們從大往小，從內往外考慮填數。
首先最值的位置必定是三個面的交點，而且最值必定是當前全部剩餘可填的數中的最大值。因此只須要肯定剩下的數的數就好了，這個時候還有\(v[i]-1\)個數能夠選，要選\(v[i]-v[i-1]-1\)個數，因此填進去的方案數就是\(\frac{(v[i]-1)!}{v[i-1]!}\)。
因此\(h[k]=h[k-1]*\frac{(v[k]-1)!}{v[k-1]!}\)。
因此\(\displaystyle h[k]=\prod_{i=1}^{k} \frac{(v[i]-1)!}{v[i-1]!}\)
而後把答案式掏出來，是：
\[\begin{aligned} &\ \ \ \ \displaystyle {V\choose v[k]}w[k](V-v[k])!h[k]\\ &=\frac{V!}{v[k]!}w[k]h[k]\\ &=V!\frac{1}{v[k]!}w[k]h[k]\\ &=V!\frac{1}{v[k]!}w[k]\prod_{i=1}^k(v[i]-1)!\prod_{i=0}^{k-1}\frac{1}{v[i]!}\\ &=V!w[k]\prod_{i=1}^k \frac{1}{v[i]} \end{aligned}\]
而後題目要求的是機率，因此和\(V!\)就沒有關係了。
那麼就只有後半部分。
這樣子就很容易計算了。

#include<iostream>
#include<cstdio>
using namespace std;
#define MAX 5001000
#define MOD 998244353
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[MAX],jv[MAX],inv[MAX];
int n,m,l,V,M,k,ans;
int v[MAX],w[MAX],s[MAX],invs[MAX];
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 C(int n,int m){return 1ll*jc[n]*jv[m]%MOD*jv[n-m]%MOD;}
int main()
{
    jc[0]=jv[0]=inv[0]=inv[1]=1;
    for(int i=2;i<MAX;++i)inv[i]=1ll*inv[MOD%i]*(MOD-MOD/i)%MOD;
    for(int i=1;i<MAX;++i)jc[i]=1ll*jc[i-1]*i%MOD;
    for(int i=1;i<MAX;++i)jv[i]=1ll*jv[i-1]*inv[i]%MOD;
    int T=read();
    while(T--)
    {
        n=read();m=read();l=read();k=read();V=1ll*n*m%MOD*l%MOD;M=min(min(n,m),l);ans=0;
        for(int i=1;i<=M;++i)v[i]=(V-1ll*(n-i)*(m-i)%MOD*(l-i)%MOD+MOD)%MOD;
        for(int i=1;i<=M;++i)w[i]=1ll*C(n,i)*C(m,i)%MOD*C(l,i)%MOD*jc[i]%MOD*jc[i]%MOD*jc[i]%MOD;
        s[0]=1;for(int i=1;i<=M;++i)s[i]=1ll*s[i-1]*v[i]%MOD;
        invs[M]=fpow(s[M],MOD-2);for(int i=M-1;i;--i)invs[i]=1ll*invs[i+1]*v[i+1]%MOD;
        for(int i=k,d=1;i<=M;++i,d=MOD-d)ans=(ans+1ll*d*C(i,k)%MOD*w[i]%MOD*invs[i])%MOD;
        printf("%d\n",ans);
    }
    return 0;
}