【CTS2019】珍珠【生成函数，二项式反演】

题目链接：洛谷

---

pb大佬说这是sb题感觉好像有点过fan。。。（我还是太弱了）

首先，设$i$这个数在序列中出现$a_i$次，要求$sum_{i=1}^D[a_i mod 2]leq n-2m$

如果要直接计算$leq n-2m$的数量会非常麻烦，所以考虑设$g_i$表示恰好出现$i$个奇数的方案之和。

这样也还是太麻烦，我们考虑使用反演或容斥通过$geq i$的数量推算出恰好等于$i$的数量，假设$f_i$表示出现$i$个奇数的方案数。

因为这是数的排列问题，所以考虑使用指数型生成函数。

$$egin{align*}f_k&=[x^n]n!inom{D}{k}(frac{e^x-e^{-x}}{2})^k(e^x)^{D-k} \&=[x^n]frac{n!}{2^k}inom{D}{k}(e^x-e^{-x})^ke^{(D-k)x}end{align*}$$

前面系数的部分可以不用管它，直接看后面多项式的部分。

$$egin{align*}&(e^x-e^{-x})^ke^{(D-k)x}[x^n] \=&sum_{i=0}^kinom{k}{i}e^{ix}e^{-(k-i)x}(-1)^{k-i}e^{(D-k)x}[x^n] \=&sum_{i=0}^kinom{k}{i}e^{(2i+D-2k)x}(-1)^{k-i}[x^n] \=&sum_{i=0}^kinom{k}{i}e^{(D-2i)x}(-1)^{k-i}[x^n] \=&frac{1}{n!}sum_{i=0}^kinom{k}{i}(-1)^i(D-2i)^nend{align*}$$

$$f_k=frac{D!}{2^k(D-k)!}sum_{i=0}^kfrac{(-1)^i(D-2i)^n}{i!}*frac{1}{(k-i)!}$$

可以通过卷积$O(Dlog D)$计算。

计算$g_k$可以使用二项式反演，熟悉二项式反演的同学可以知道，二项式反演其实就是指数型生成函数的应用。
$$f_i=sum_{j=i}^ninom{j}{i}g_jRightarrow g_i=frac{1}{i!}sum_{j=k}^nfrac{(-1)^{j-i}}{(j-i)!}*(f_j*j!)$$

$$ans=sum_{i=0}^{n-2m}g_i$$

时间复杂度$O(Dlog D)$

#include<bits/stdc++.h>
#define Rint register int
using namespace std;
typedef long long LL;
const int N = 800003, mod = 998244353, g = 3, gi = 332748118;
inline int kasumi(int a, int b){
    int res = 1;
    while(b){
        if(b & 1) res = (LL) res * a % mod;
        a = (LL) a * a % mod;
        b >>= 1;
    }
    return res;
}
int fac[N], inv[N], invfac[N], ans;
inline void init(int n){
    fac[0] = 1;
    for(Rint i = 1;i <= n;i ++) fac[i] = (LL) fac[i - 1] * i % mod;
    invfac[n] = kasumi(fac[n], mod - 2);
    for(Rint i = n;i;i --){
        invfac[i - 1] = (LL) i * invfac[i] % mod;
        inv[i] = (LL) invfac[i] * fac[i - 1] % mod;
    }
}
int rev[N];
inline int calrev(int n){
    int limit = 1, L = -1;
    while(limit <= n){limit <<= 1; L ++;}
    for(Rint i = 0;i < limit;i ++)
        rev[i] = (rev[i >> 1] >> 1) | ((i & 1) << L);
    return limit;
}
inline void NTT(int *A, int limit, int type){
    for(Rint i = 0;i < limit;i ++)
        if(i < rev[i]) swap(A[i], A[rev[i]]);
    for(Rint mid = 1;mid < limit;mid <<= 1){
        int Wn = kasumi(type == 1 ? g : gi, (mod - 1) / (mid << 1));
        for(Rint j = 0;j < limit;j += (mid << 1)){
            int w = 1;
            for(Rint k = 0;k < mid;k ++, w = (LL) w * Wn % mod){
                int x = A[j + k], y = (LL) w * A[j + k + mid] % mod;
                A[j + k] = (x + y) % mod;
                A[j + k + mid] = (x - y + mod) % mod;
            }
        }
    }
    if(type == -1){
        int inv = kasumi(limit, mod - 2);
        for(Rint i = 0;i < limit;i ++) A[i] = (LL) A[i] * inv % mod;
    }
}
int D, n, m, F[N], G[N];
int main(){
    scanf("%d%d%d", &D, &n, &m);
    if(n < 2 * m){
        printf("0");
        return 0;
    }
    if(D <= n - 2 * m){
        printf("%d", kasumi(D, n));
        return 0;
    }
    init(D);
    for(Rint i = 0;i <= D;i ++){
        F[i] = (LL) kasumi((D - 2 * i + mod) % mod, n) * invfac[i] % mod;
        if(i & 1) F[i] = mod - F[i];
        G[i] = invfac[i];
    }
    int limit = calrev(D << 1);
    NTT(F, limit, 1); NTT(G, limit, 1);
    for(Rint i = 0;i < limit;i ++) F[i] = (LL) F[i] * G[i] % mod;
    NTT(F, limit, -1);
    for(Rint i = 0;i < limit;i ++) G[i] = 0;
    for(Rint i = D + 1;i < limit;i ++) F[i] = 0;
    for(Rint i = 0;i <= D;i ++){
        F[i] = (LL) F[i] * fac[i] % mod * fac[D] % mod * kasumi(2, mod - 1 - i) % mod * invfac[D - i] % mod;
        G[D - i] = (i & 1) ? (mod - invfac[i]) : invfac[i];
    }
    NTT(F, limit, 1); NTT(G, limit, 1);
    for(Rint i = 0;i < limit;i ++) F[i] = (LL) F[i] * G[i] % mod;
    NTT(F, limit, -1);
    for(Rint i = 0;i <= n - 2 * m;i ++) ans = (ans + (LL) F[i + D] * invfac[i] % mod) % mod;
    printf("%d", ans);
}