【训练题64：组合数学 | 容斥】Product | 2021牛客暑期多校训练营4

题意

• Product | 2021牛客暑期多校训练营4
  给你 $n,k,D$，你需要求出：
  $$\sum _{D=\sum a_i,a_i\ge 0}\frac{D!}{\prod (a_i+k)!}$$
• $1\le n\le 50$
  $0\le k\le 50$
  $0\le D\le 10^8$

思路

• 对现在的我来说也是神仙题，先学着点…
  根据组合数学，我们有这个式子：
  $$\sum _{D=\sum a_i,a_i\ge 0}\frac{D!}{\prod (a_i)!}=n^D$$
  为什么呢？左边的式子就相当于，有 $D$ 个球，放进 $n$ 个箱子里，球和箱子都不同的方案数，使用的是多重集的全排列公式 （因为一个集合里放进小球 $AB$ 与小球 $BA$ 当然是同一种方案，多了 $a_i!$ 的方案数）
  而等式右边就是，对于每一个球，放进这 $n$ 个集合中的答案都不相同，所以就是 $n^D$
• 进而，我们可以修改成和题意类似的式子：
  $$\sum _{D+nk=\sum a_i,a_i\ge 0}\frac{(D+nk)!}{\prod (a_i)!}=n^{D+nk}$$
  但是其实题意的式子是要我们求这样子：
  $$Ans=\sum _{D+nk=\sum a_i,a_i\ge k}\frac{(D)!}{\prod (a_i)!}$$
  我们可以先稍微转换一下，变成这样：
  $$Ans=\frac{D!}{(D+nk)!}\sum _{D+nk=\sum a_i,a_i\ge k}\frac{(D+nk)!}{\prod (a_i)!}\text{\hspace{1em}(1)}$$
  现在的问题就是，我们要求的是 $a_i\ge k$ ，但是我们的式子只支持我们 $a_i\ge 0$
  这个时候就要用到容斥了！（怎么又是你…）
• 我们定义合法的方法，就是这个集合中放了 $\ge k$ 个球；非法就是集合中放了 $<k$ 个球
  设 $dp[i][j]$ 表示前 $i$ 个集合，共放了 $j$ 个球，且全部都是非法集合的方案数
  这个转移是这样子的：
  $$dp[i][j]=\sum _{t=0}^{k-1}dp[i-1][j-t]\times C_j^t$$
  就是这 $j$ 个球中，枚举时哪 $t$ 个球放在第 $i$ 个集合中
• 接下来算我们的答案，枚举 $n$ 组中有 $i$ 组不合法，这 $i$ 组中用了 $j$ 个球，方案就是
  $$dp[i][j]\times C_n^i\times C_{D+nk}^j$$
  那么合法的组数就是 $n-i$ 组，共用了 $D+nk-j$ 个球，根据我们之前的式子，方案数就是：
  $$(n-i{)}^{D+nk-j}$$
  再加上容斥的内容，还有我们 (1) 式中最左边的系数，就得到了：
  $$Ans=\frac{D!}{(D+nk)!}\times \sum _{i=0}^n(-1{)}^i\sum _{j=0}^{(i-1)k}(dp[i][j]\times C_n^i\times C_{D+nk}^j)\times ((n-i{)}^{D+nk-j})$$

代码

• 时间复杂度：$O((nk{)}^2)$ 主要是算 $dp[i][j]$ 的时候的

#include <bits/stdc++.h>
#define IOS ios::sync_with_stdio(false);cin.tie(NULL);cout.tie(NULL);
using namespace std;
typedef long long ll;
void show(){std::cerr << endl;}template<typename T,typename... Args>void show(T x,Args... args){std::cerr << "[ " << x <<  " ] , ";show(args...);}

const int MAX = 1e5+50;
const int MOD = 998244353;
const int INF = 0x3f3f3f3f;
const ll LINF = 0x3f3f3f3f3f3f3f3f;
const double EPS = 1e-5;

ll qpow(ll a,ll n){/* */ll res = 1LL;while(n){if(n&1)res=res*a%MOD;a=a*a%MOD;n>>=1;}return res;}
ll qpow(ll a,ll n,ll p){a%=p;ll res = 1LL;while(n){if(n&1)res=res*a%p;a=a*a%p;n>>=1;}return res;}
ll npow(ll a,ll n){/* */ll res = 1LL;while(n){if(n&1)res=res*a;a=a*a;n>>=1;if(res<0||a<0)return 0;}return res;}
ll inv(ll a){/* */return qpow(a,MOD-2);}
ll inv(ll a,ll p){return qpow(a,p-2,p);}

ll dp[55][3500];
ll fac[3500],ivfac[3500];

void init(int x){
    fac[0] = 1;
    for(int i = 1;i <= x;++i)fac[i] = fac[i-1] * i % MOD;
    ivfac[x] = inv(fac[x]);
    for(int i = x - 1;~i;--i)ivfac[i] = ivfac[i+1] * (i+1) % MOD;
}
ll C(int n,int m){
    if(n < 0 || m > n)return 0;
    return fac[n] * ivfac[m] % MOD * ivfac[n - m] % MOD;
}
int main()
{
    init(2500);
    ll n,k,D;cin >> n >> k >> D;

    dp[0][0] = 1;
    for(int i = 1;i <= n;++i){
        for(int j = 0;j <= (k - 1) * i;++j){
            for(int t = 0;t <= k - 1;++t){
                if(j-t>=0)dp[i][j] = (dp[i][j] + dp[i-1][j-t] * C(j,t) % MOD) % MOD;
            }
        }
    }

    ll ans = 0;

    for(int i = 0;i <= n;++i){
        ll pm = (i&1) ? -1 : 1;
        ll tmp = 1;
        for(int j = 0;j <= i * (k - 1);++j){
            ans = (ans + pm * dp[i][j] * C(n,i) % MOD * tmp % MOD * qpow(n-i,D+n*k-j) % MOD + MOD) % MOD;
            tmp = tmp * inv(j+1) % MOD * (D + n * k - j) % MOD;
        }
    }
    for(int i = 1;i <= n * k;++i){
        ans = ans * inv(D + i) % MOD;
    }
    cout << ans;
    return 0;
}