牛客2021多校第四场G（生成函数， 组合数学，计数） 题解

我最近怎么总是做生成函数的题目

题目大意

对于给定的$n,k,D$​, 求${\sum }_{[\sum a_i=D]}\frac{D!}{\prod (a_i+k)!}$​

若答案表示为$\frac{x}{y}$​ 则输出$\frac{x}{y}mod998244353$​​

其中
$$\begin{gathered} 1\le n\le 50 \\ 0\le k\le 50 \\ 0\le D\le 10^8 \end{gathered}$$
我是链接

题目分析

前置芝士（结论）

$$\sum _{[{\sum }_1^n{a}_i=D]}\frac{D!}{{\prod }_1^n{a}_i!}=n^D$$

这个结论理解为在$[1..n]$​​中选数, 组成长度为$D$​​的数列的方案数, $a_i$​ 表示$i$在这个数列中出现的次数

$D!$​​​是全排列, 除掉的$\prod a_i!$​​​是对于每一个数的重复方案数, $\frac{D!}{{\prod }_1^n{a}_i}$​​​是对于一个特定数列$a_i$​​​的产生的方案数，求和便得到了所有组合的方案数。所有组合的方案数必定是每一位都是$n$​种可能性,所以求和得到了$n^D$​

因此可以得知
$$\sum _{[{\sum }_1^n{a}_i=D]}\frac{1}{{\prod }_1^n{a}_i!}=\frac{n^D}{D!}$$
而这个问题使用生成函数解释为

构造指数型母函数$g_k(x)={\sum }_0^{\infty }\frac{x^i}{i!}=e^x$​​表示对于第$k$​个数字选取了$i$次

那么总共的方案数就有$f(x)={\prod }_1^n{g}_k(x)$其中第$D$项的系数就表示从$[1..n]$中生成序列的方案数，就是上文中的$\frac{n^D}{D!}$

所以$f(x)=(e^x{)}^n$​

$f(x)$的第$D$项的系数为$\frac{n^D}{D!}$

正经分析

本题题目可以改写为
$$\begin{gathered} \sum _{[\sum a_i=D+nk,a_i>k]}\frac{D!}{\prod (a_i)!} \\ =D!\ast \sum _{[\sum a_i=D+nk,a_i>k]}\frac{1}{\prod (a_i)!} \end{gathered}$$
发现将$D!$​​提出后剩下的部分由上文的解释方法为从$[1..n]$​​中选取数字,每个数字至少选取$k$​​次,生成总长度为$D+nk$​​的序列,求总方案数

所以这部分使用生成函数来解释
$$\begin{gathered} g_k(x)=\sum _k^{\infty }\frac{x^i}{i!}=e^x-\sum _0^{k-1}\frac{x^i}{i!} \\ f(x)=\prod _1^n{g}_k(x)=(e^x-\sum _0^{k-1}\frac{x^i}{i!}{)}^n \\ f(x)=e^{nx}-(\genfrac{}{}{0ex}{}{n}{1})e^{(n-1)x}(\sum _0^{k-1}\frac{x^i}{i!})+... \\ =\sum _0^n(-1{)}^i\left(\genfrac{}{}{0ex}{}{n}{i}\right)e^{(n-i)x}(\sum _0^{k-1}\frac{x^j}{j!}{)}^i \end{gathered}$$
剩下的部分就是$f(x)$在$x^{D+nk}$处的系数

所以怎么才能求出这个系数$S$呢

如果$({\sum }_0^{k-1}\frac{x^j}{j!}{)}^i$​​​​​展开的各项系数为$K_{i,j}$​​​​​​, 而$e^{(n-i)x}$​在$j$处的系数为$S_{i,j}$​

那么
$$S=\sum _0^n(-1{)}^i(\genfrac{}{}{0ex}{}{n}{i})(\sum _0^{i(k-1)}{S}_{i,D+nk-j}{K}_{i,j})$$
我们发现$S_{i,j}=\frac{(n-i{)}^j}{j!}$ 也就是说$S_{i,D+nk-j}=\frac{(n-i{)}^{(D+nk-j)}}{(D+nk-j)!}$

而$K_{i,j}$可以由递推的方式求出
$$\begin{gathered} (K_{i-1,0}+K_{i-1,1}x+...+K_{i-1,i(k-1)}{x}^{i(k-1)})(\sum _0^{k-1}\frac{x^j}{j!}) \\ \therefore K_{i,j}=\sum _{x=0}^{k-1}{K}_{i-1,k-x}\ast \frac{1}{x!},j\in [0,i(k-1)] \end{gathered}$$
有了递推式，就可以使用DP的方式求出$K_{i,j}$了

可是这个题目并没有结束，1e8的范围求若干次阶乘，时间复杂度无法接受，因此将$D!$​​乘进去,与$S_{i,D+nk-j}$​​结合, 得到$\frac{(n-i{)}^{(D+nk-j)}D!}{(D+nk-j)!}$​将$\frac{D!}{(D+nk-j)!}$结合计算，发现每次最多只需要计算$nk$​个数相乘，时间变得非常充裕

综上
$$ANS=D!\ast S=\sum _0^n(-1{)}^i(\genfrac{}{}{0ex}{}{n}{i})(\sum _0^{i(k-1)}(n-i{)}^{D+nk-j}\frac{D!}{(D+nk-j)!}{K}_{i,j})$$

由此我们解决了求$S$的若干问题,DP求$K$的复杂度为$O((nk{)}^2)$ 最后统计答案的时间复杂度为$O(n^{2}k)$

所以最后的时间复杂度为$O((nk{)}^2)$

通过了这个题目

通过代码

人比较菜，代码比较丑，dalao勿喷

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cmath>
#include <algorithm>
using namespace std;
const int MAXN = 50;
const int MOD = 998244353;
typedef long long ll;
ll dp[MAXN + 10][MAXN * MAXN + 10];
ll qp(ll a, int b)
{
    ll ans = 1;
    while(b)
    {
        if(b & 1)
            ans = (ans * a) % MOD;
        b >>= 1;
        a = (a  * a) % MOD;
    }
    return ans;
}
ll inv(ll a)
{
    return qp(a, MOD - 2);
}
ll fac[50 * 50 + 100];
ll facInv[50 * 50 + 100];
ll facDInv[50 * 50 + 100];
ll C(int n, int m)
{
    return (((fac[n] * facInv[m]) % MOD) * facInv[n - m]) % MOD;
}
int main()
{
    int n,k,d;
    scanf("%d%d%d", &n, &k, &d);
    
    fac[0] = 1;
    for(int i = 1; i <= max(n * k, max(n, k)); ++i)
        fac[i] = (fac[i - 1] * i) % MOD;
    for(int i = 0; i <= max(n * k, max(n, k)); ++i)
        facInv[i] = inv(fac[i]);
    //dp 用来计算$$(e^x-\sum{i=0}{k-1}\frac{x^i}{i!})^n$$的各项系数中非e^?的部分
    dp[0][0] = 1;
    for(int i = 1; i <= n; ++i)
    {
        for(int j = 0; j <= i * (k - 1); ++j)
        {
            for(int x = 0; x <= min(j, k - 1); ++x)
            {
                dp[i][j] += (dp[i - 1][j - x] * facInv[x]) % MOD;
                dp[i][j] %= MOD;
            }
            
        }
    }
    facDInv[0] = 1;
    for(int i = 1; i <= n * k; ++i)
        facDInv[i] = (facDInv[i - 1] * ((d + i) % MOD)) % MOD; 
    for(int i = 1; i <= n * k; ++i)
        facDInv[i] = inv(facDInv[i]);
    ll ans = 0;
    int sgn = -1;
    for(int i = 0; i <= n; ++i)
    {
        sgn *= -1;
        ll tmp = 0;
        for(int j = 0; j <= i * (k - 1); ++j)
        {
            tmp +=(((dp[i][j] * qp(n - i, d + n * k - j)) % MOD) * facDInv[n * k - j]) % MOD;
            tmp %= MOD;
        }
        tmp *= C(n, i);
        tmp %= MOD;
        tmp *= sgn;
        tmp %= MOD;
        ans += tmp;
        ans %= MOD;
    }
    ans += MOD;
    ans %= MOD;
    printf("%lld\n", ans);
    return 0;
}