LOJ3119. 「CTS2019 | CTSC2019」随机立方体 二项式反演

题目传送门

https://loj.ac/problem/3119

现在 BZOJ 的管理员已经不干活了吗,CTS(C)2019 和 NOI2019 的题目到现在还没与传上去。

果然还是 LOJ 好。

题目

恰好有 \(k\) 个极大数不太好求,我们还是转化成二项式反演。

然后就变成了给定一个点的集合 \(S\),求钦定 \(S\) 中的点是极大点的方案数。可以发现 \(S\) 中的点因为必须要保证没有一维的坐标相同,所以到底是哪些点是不重要的,有用的只有 \(|S|\)。所以问题转化为钦定了 \(k\) 个点必须是极大点的方案数。

我们可以发现一个性质,如果我们把所有的极大点的数字从小到大考虑的话,那么每次选完和当前个点有关的的面上的数字以后,剩下的点上的数字如何安排实际上是一个 \((n-1)\times(m-1)\times(l-1),k-1\) 的子问题了。

我们需要先选出 \(k\) 个被我们钦定的点的排列,它们应该满足没有一维的坐标是相同的,于是就很显然是 \(\binom nk\binom mk\binom lk(k!)^3\)。至于为什么是排列,上面说了,只有从小到大考虑,才满足上面的性质。

然后我们需要给被我们选出来的 \(k\) 个点所在的所有面上的所有位置、剩下来的所有位置安排数字了。剩下来的 \(nml-(n-k)(m-k)(l-k)\) 很好安排,先从所有的 \(nml\) 个数字中选出来那么多,然后自由排列一下就好了。这个是 \(\binom{nml}{(n-k)(m-k)(l-k)}((n-k)(m-k)(l-k))!\)

主要的难点在于被选出的 \(k\) 个点所在的面上的数字。我们有 \(nmk - (n-k)(m-k)(l-k)\) 个备选数字。因为我们是从小到大排列的,所以被最后选出来的位置上的数字应该是最大的。同时,它还会带走 \((n-k+1)(m-k+1)(l-k+1) - (n-k)(m-k)(l-k)\) 个最后选它的时候的子立方体中和它同一面的数字。此时,因为它是最大的,所以无论选哪些数字都不会影响前一位置的选择。因此它有 \(\frac{(nml-(n-k)(m-k)(l-k)-1)!}{(nml-(n-k+1)(m-k+1)(l-k+1))!}\) 中选择方法。

然后选完以后,倒数第二个被选出来的位置上的数字,应该是剩下的数字中最大的。而剩下的数字一共有 \(nml-(n-k+1)(m-k+1)(l-k+1)\) 种,于是和之前最后一个一样一样考虑就可以了。

因此给被选出的 \(k\) 个点所在的面上安排数字的总方案数为
\[ \prod_{i=1}^k \frac{(nml-(n-i)(m-i)(l-i)-1)!}{(nml-(n-i+1)(m-i+1)(l-i+1))!} \]
可以发现在阶乘上的我们可以消掉一部分,于是就是
\[ (nml-(n-k)(m-k)(l-k)-1)!\prod_{i=1}^{k-1} \frac 1{(nml-(n-i+1)(m-i+1)(l-i+1))} \]
于是把上面的东西都整合一下,钦定了 \(k\) 个点必须是极大点的方案数为
\[ \binom nk\binom mk\binom lk(k!)^3\binom{nml}{(n-k)(m-k)(l-k)}((n-k)(m-k)(l-k))!(nml-(n-k)(m-k)(l-k)-1)!\prod_{i=1}^{k-1} \frac 1{(nml-(n-i+1)(m-i+1)(l-i+1))} \]

扫描二维码关注公众号,回复: 7258583 查看本文章

\[ =\binom nk\binom mk\binom lk(k!)^3\frac{(nml)!}{((n-k)(m-k)(l-k))!(nml-(n-k)(m-k)(l-k))!}((n-k)(m-k)(l-k))!(nml-(n-k)(m-k)(l-k)-1)!\prod_{i=1}^{k-1} \frac 1{(nml-(n-i+1)(m-i+1)(l-i+1))} \]

\[ = \binom nk\binom mk\binom lk(k!)^3{(nml)!}\prod_{i=1}^{k} \frac 1{(nml-(n-i+1)(m-i+1)(l-i+1))} \]
\(f(k)\) 等于上面那个钦定了 \(k\) 个点必须是极大点的方案数,\(g(k)\) 表示恰好 \(k\) 个点的方案数,那么显然有
\[ f(k)=\sum_{i=k}^n\binom ikg(i) \]
所以直接二项式反演就可以了。

因为是求期望,最后记得除掉 \((nml)!\)


下面是代码。\(\prod\limits_{i=1}^{k} \frac 1{(nml-(n-i+1)(m-i+1)(l-i+1))}\) 可以线性预处理,因此总的时间复杂度为 \(O(n)\)

#include<bits/stdc++.h>

#define fec(i, x, y) (int i = head[x], y = g[i].to; i; i = g[i].ne, y = g[i].to)
#define dbg(...) fprintf(stderr, __VA_ARGS__)
#define File(x) freopen(#x".in", "r", stdin), freopen(#x".out", "w", stdout)
#define fi first
#define se second
#define pb push_back

template<typename A, typename B> inline char smax(A &a, const B &b) {return a < b ? a = b , 1 : 0;}
template<typename A, typename B> inline char smin(A &a, const B &b) {return b < a ? a = b , 1 : 0;}

typedef long long ll; typedef unsigned long long ull; typedef std::pair<int, int> pii;

template<typename I>
inline void read(I &x) {
    int f = 0, c;
    while (!isdigit(c = getchar())) c == '-' ? f = 1 : 0;
    x = c & 15;
    while (isdigit(c = getchar())) x = (x << 1) + (x << 3) + (c & 15);
    f ? x = -x : 0;
}

const int N = 5e6 + 7;
const int P = 998244353;

int n, m, l, k;
int a[N], s[N], f[N];

inline int smod(int x) { return x >= P ? x - P : x; }
inline void sadd(int &x, const int &y) { x += y; x >= P ? x -= P : x; }
inline int fpow(int x, int y) {
    int ans = 1;
    for (; y; y >>= 1, x = (ll)x * x % P) if (y & 1) ans = (ll)ans * x % P;
    return ans;
}

int fac[N], inv[N], ifac[N];
inline void ycl(const int &n = ::n) {
    fac[0] = 1; for (int i = 1; i <= n; ++i) fac[i] = (ll)fac[i - 1] * i % P;
    inv[1] = 1; for (int i = 2; i <= n; ++i) inv[i] = (ll)(P - P / i) * inv[P % i] % P;
    ifac[0] = 1; for (int i = 1; i <= n; ++i) ifac[i] = (ll)ifac[i - 1] * inv[i] % P;
}

inline int C(int x, int y) {
    if (x < y) return 0;
    return (ll)fac[x] * ifac[y] % P * ifac[x - y] % P;
}

inline void work() {
    ycl(std::max(m, l));
    int nml = (ll)n * m % P * l % P, ii = 1;
    a[0] = 1;
    for (int i = 1; i <= n; ++i) a[i] = (nml + P - (ll)(n - i) * (m - i) % P * (l - i) % P) % P, ii = (ll)ii * a[i] % P;
    ii = fpow(ii, P - 2);
    for (int i = n; i; --i) s[i] = ii, ii = (ll)ii * a[i] % P;
    for (int i = 1; i <= n; ++i) f[i] = (ll)C(n, i) * C(m, i) % P * C(l, i) % P * fpow(fac[i], 3) % P * s[i] % P;
    int ans = 0;
    for (int i = k; i <= n; ++i)
        if ((i - k) & 1) sadd(ans, P - (ll)C(i, k) * f[i] % P);
        else sadd(ans, (ll)C(i, k) * f[i] % P);
    printf("%d\n", ans);
}

inline void init() {
    read(n), read(m), read(l), read(k);
    if (n > m) std::swap(n, m);
    if (n > l) std::swap(n, l);
}

int main() {
#ifdef hzhkk
    freopen("hkk.in", "r", stdin);
#endif
    int T;
    read(T);
    while (T--) {
        init();
        work();
    }
    fclose(stdin), fclose(stdout);
    return 0;
}

猜你喜欢

转载自www.cnblogs.com/hankeke/p/LOJ3119.html