2021牛客多校四 G_product

最新推荐文章于 2023-10-14 05:08:33 发布

juraws

最新推荐文章于 2023-10-14 05:08:33 发布

阅读量130

点赞数

分类专栏： # 2021牛客多校

本文链接：https://blog.csdn.net/qq_39602052/article/details/119495123

版权

2021牛客多校专栏收录该内容

4 篇文章 0 订阅

订阅专栏

G_product

题意： 给出n, k, D. 求 $\frac{D!}{\prod_{i = 1}^{n}(a_i + k)!} （\sum_{i=1}^{n} a_i = D）$

G-Product_2021牛客暑期多校训练营4 (nowcoder.com)

参考了欣君的题解。【算法题解】2021牛客暑期多校训练营#4_哔哩哔哩_bilibili

首先当 $k = = 0$ （没有k）有个结论：
$\sum_{a_i \geq 0, \sum a_i = D} \prod_i \frac{1}{a_i} = \frac{n^D}{D!}$

这里可以先考虑这个式子： $\frac{D!}{\prod_{i = 1}^{n}a_i!} = n ^D$ ，问题转化为D个球染1-n种颜色， $a_1 \dots a_n$ 表示染了每种颜色的数量

$\Rightarrow D! \sum_{a_i \geq k, \sum a_i = D+nk} \prod_i \frac{1}{a_i!}$

先计算这个式子：
$\sum_{a_i \geq 0, \sum a_i = D+nk} \prod_i \frac{1}{a_i!} \\ =\frac{n^{D +nk}}{(D+nk)!} D!$
再将 $a_i < k$ 的部分容斥掉：

考虑哪些 $a_i < k$ ，枚举 $a_i$ 的值，但 n, D 不同，需要DP

设 $dp_{i, j}$ 表示将j个不同的球分成i组，每组不超过k-1个的方案数， $dp_{i, j} = \sum _{t = 0}^{k - 1} dp_{i - 1, j - t} \tbinom{t}{j}$ ，（这部分可以预处理，虽然D很大，但n ,k 很小）

注意转化后的 $a_i$ 是 $\ge k$ 的 ,

枚举有i个不合法的 $a_i$ ，这些 $a_i$ 的和为j，再对这部分分配一下，可以得到 $\tbinom{i}{n} \tbinom{j}{D + nk} dp_{i , j}$

剩余的都满足 $a_i \ge k$ ，这部分可以用隔板法直接求，

一共有n - i 组， D+nk - j 个球，每组球超过k个的方案数

$\tbinom{n- i - 1}{D- j}$

D很大好像不大好处理（

剩余的部分可以任意取，于是直接列一下就是$ (n - i) ^ {D+ nk - j}$，因为会有重复计算，这里要容斥一下 $1)^{i}$

结合一下： $\sum_{i} \sum_{j}(- 1)^{i} \tbinom{i}{n} \tbinom{j}{D + nk} dp_{i , j} (n - i) ^ {D+ nk - j}$

然后把 $\frac{D!}{(D+nk)!}$ 乘上去

#include <bits/stdc++.h>

using namespace std;

typedef long long ll;
const ll N = 60;
const ll M = 2510;
const ll mod = 998244353;

ll dp[N][M], c[M][M], cc[M];

ll n, d, k;

inline ll qpow(ll x, ll p) {
    ll res = 1;
    while (p) {
        if (p & 1) res = (res * x) % mod;
        x = (x * x) % mod;
        p >>= 1;
    }
    return res;
}

inline void init() {
    for (ll i = 0; i < M; ++i) {
        c[i][0] = 1;
        for (ll j = 1; j <= i; ++j) {
            c[i][j] = (c[i - 1][j] + c[i - 1][j - 1]) % mod;
        }
    }
    dp[0][0] = 1;
    for (ll i = 1; i < N; ++i) {
        for (ll j = 0; j < M; ++j) {
            for (ll t = 0; t < k; ++t) {
                if(j < t) continue;
                dp[i][j] = (dp[i][j] + dp[i - 1][j - t] * c[j][t]% mod) % mod;
            }
        }
    }
}

signed main() {
    ios::sync_with_stdio(false), cin.tie(nullptr), cout.tie(nullptr);

    cin >> n >> k >> d;

    init();

    ll ans = 0;
    cc[0] = 1;
    for (ll i = 1; i < M; ++i) {
        cc[i] = cc[i - 1] * (d + n * k - i + 1 + mod) % mod * qpow(i, mod - 2) % mod;
    }

    for (ll i = 0; i <= n; ++i) {
        ll tmp = 0;
        for (ll j = 0; j <= i * (k - 1); ++j) {
            tmp = (tmp + cc[j] * dp[i][j] % mod * qpow(n - i, d + n * k - j) % mod) % mod;
        }
        tmp = tmp * c[n][i] % mod;
        if (i & 1) ans = (ans - tmp + mod) % mod;
        else ans = (ans + tmp) % mod;
    }

    for (ll i = d + 1; i <= d + n * k; ++i) {
        ans = ans * qpow(i, mod - 2) % mod;
    }
    cout << ans << endl;
    return 0;
}