2021牛客多校#4 G-Product

原题链接

				https://ac.nowcoder.com/acm/contest/11255/G

题目大意

给定三个整数$n,k,D(1\le n,k\le 50,0\le D\le 10^8)$。
定义非负整数序列$a$的权值为$$\frac{D!}{\prod _{i=1}^n(a_i+k)!}$$
求所有满足以下条件的非负整数序列$a$的权值和：
1.$\forall \in [1,n],a_i\ge 0$
2.${\sum }_{i=1}^n{a}_i=D$
答案对$998244353$取模。

题解

我们可以把题目转化成把$D$个小球分成$n$组，每组$a_i$个球。

①我们先不考虑$+k$的情况，将$D$个不同的求分为$n$组，每个球有$n$种分组可能，总共方案数为$n^D$
$$\sum _{a_i\ge 0,\sum _{i=1}^n{a}_i=D}\frac{D!}{\prod _{i=1}^n{a}_i!}$$
则$(D+nk)$个球分$n$组的总方案书为$n^{D+nk}$。

②将$+k$合并入$a_i$中：
$$\sum _{ai\ge k,\sum _{i=1}^n{a}_i=D+nk}\frac{D!}{{\prod }_{i=1}^n{a}_i!}$$

将②中$a_i\ge k$换成$a_i\ge 0$，与①可推得总方案数(包括不合理方案)：
$$\frac{D!}{(D+nk)!}{n}^{D+nk}$$

设$d{p}_{i,j}$表示$j$个不同的求分为$i$个非法组（即魅族的球数都小于$k$），通过美剧最后一个非法组的球数与球的分配情况，得转移式：
$$d{p}_{i,j}=\sum _{t=0}^{k-1}d{p}_{i-1,j-t}{C}_j^t$$

通过计算，得出最终转移式：
$$\frac{D!}{(D+nk)}!\sum _{i=0}^n(-1{)}^i\sum _{j=0}^{i(k-1)}d{p}_{i,j}\times C_n^i\times C_{D+nk}^j\times d{p}_{i,j}(n-i{)}^{D+nk-j}$$

参考代码

#include<bits/stdc++.h>
#define ll long long
using namespace std;
const int N=55;
const int MOD=998244353;
ll powmod(ll a,ll b)
{
    ll ret=1;
    while(b)
    {
        if(b&1)ret=ret*a%MOD;
        a=a*a%MOD;
        b>>=1;
    }
    return ret;
}
ll C[N*N][N*N],dp[N][N*N];
ll n,k,D;
void init()
{
    for(int i=0;i<N*N;i++)
        C[i][0]=1;
    for(int i=1;i<N*N;i++)
        for(int j=1;j<=min(N,i);j++)
            C[i][j]=(C[i-1][j]+C[i-1][j-1])%MOD;
    dp[0][0]=1;
    for(int i=0;i<n;i++)
        for(int j=0;j<=i*(k-1);j++)
            for(int t=0;t<k;t++)
                dp[i+1][j+t]=(dp[i+1][j+t]+dp[i][j]*C[j+t][t])%MOD;
}
int main()
{
    scanf("%lld%lld%lld",&n,&k,&D);
    init();
    D+=n*k;
    ll ans=0;
    for(int i=0;i<=n;i++)
    {
        ll v=1;
        for(int j=0;j<=i*(k-1);j++)
        {
            ll num=(i&1?MOD-C[n][i]:C[n][i]);
            num=num*powmod(n-i,D-j)%MOD*dp[i][j]%MOD*v%MOD;
            ans=(ans+num)%MOD;
            v=v*(D-j)%MOD*powmod(j+1,MOD-2)%MOD;
        }
    }
    for(int i=D-n*k+1;i<=D;i++)
        ans=ans*powmod(i,MOD-2)%MOD;
    cout<<ans;
}