【弱省胡策】Round #5 Handle 解题报告

这个题是我出的 sb 题。

首先，我们可以得到：

$$A_i = sum_{j=i}^{n}{jchoose i}(-1)^{i+j}B_j$$

我们先假设是对的，然后我们把这个关系带进来，有：

$$B_i = sum_{j=i}^{n}{jchoose i}A_j = sum_{j=1}^{n}{jchoose i}sum_{k=j}^{n}{kchoose j}(-1)^{j+k}B_k = sum_{j=i}^{n}B_jsum_{k=i}^{j}{jchoose k}{kchoose i}(-1)^{j+k}$$

然后有：

$${jchoose k}{kchoose i} = frac{j!k!}{k!(j-k)!i!(k-i)!} = frac{j!(j-i)!}{(j-i)!(j-k)!i!(k-i)!} = {jchoose i}{j - i choose j - k}$$

故：

$$B_i = sum_{j=i}^{n}B_jsum_{k=i}^{j}{jchoose i}{j - ichoose j - k}(-1)^{j+k} = sum_{j=i}^{n}B_j{jchoose i}sum_{k=0}^{j-i}{j - ichoose k}(-1)^k$$

又因为从一个非空石子堆中选出奇数个石子和偶数个石子的方案总数是一样的，所以有：

$$sum_{i=0}^{n}{nchoose i}(-1)^i = [n == 0]$$

故：

$$B_i = sum_{j = i}^{n}B_j{jchoose i}[i == j] = B_i$$

发现我们构造的这个关系是没有矛盾的，所以就可以用这个关系来算 $A_i$ 了。

我们刚才证明了这个关系的充分性，那么必要性呢？

因为这是一个满秩的 $n$ 元一次方程组，所以有且仅有一组解满足这个关系。

然而我们现在就找到了这样的一组合法解，所以必要性也是显然的。

所以我们可以得到这两组关系互为充分必要条件，也就是等价的。

把式子展开，有：

$$A_i = sum_{j=i}^{n}frac{j!(-1)^jB_j}{i!(-1)^i(j-i)!}$$

其中 $frac{1}{i!(-1)^i}$ 我们可以最后再考虑，然后我们令 $T_i = A_{n-i} imes (n-i)! imes (-1)^{n-i}$，那么有：

$$T_i = sum_{j=n-i}^{n}frac{j!(-1)^jB_j}{(j-n+i)!} = sum_{j=0}^{i}frac{(n-j)!(-1)^{n-j}B_{n-j}}{(i-j)!}$$

令 $X_i = (n-i)!(-1)^{n - i}B_{n-i}$，$Y_i = frac{1}{i!}$，那么就有：

$$T_i = sum_{j=0}^{i}X_j imes Y_{i - j}$$

就可以用 FFT 来算了，算完之后再处理处理就 OK 了。

时间复杂度 $O(nlog n)$。

#include <cmath>
#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
typedef long long LL;
#define N 262144 + 5
#define Mod 998244353
#define g 3

int n, d, len, A[N], Fac[N], Inv[N], X[N], Y[N], Rev[N], e[N];

inline int Inc(int a, int b)
{
    return a + b - (a + b >= Mod ? Mod : 0);
}

inline int power(int u, int v)
{
    int res = 1;
    for (; v; v >>= 1)
    {
        if (v & 1) res = (LL) res * u % Mod;
        u = (LL) u * u % Mod;
    }
    return res;
}

inline void Prepare()
{
    Fac[0] = Inv[0] = 1;
    for (int i = 1; i <= n + 1; i ++)
        Fac[i] = (LL) Fac[i - 1] * i % Mod;
    Inv[n + 1] = power(Fac[n + 1], Mod - 2);
    for (int i = n; i; i --)
        Inv[i] = (LL) Inv[i + 1] * (i + 1) % Mod;
}

inline void FFT(int *p)
{
    for (int i = 0; i < len; i ++)
        if (Rev[i] > i) swap(p[i], p[Rev[i]]);
    for (int k = 1, s = 1; k < len; k <<= 1, s ++)
        for (int i = 0; i < len; i ++)
        {
            if (i & k) continue ;
            int t = (i & k - 1) << d - s;
            int u = (LL) e[t] * p[i + k] % Mod;
            p[i] = Inc(p[i], u);
            p[i + k] = Inc(p[i], Mod - Inc(u, u));
        }
}

int main()
{
    #ifndef ONLINE_JUDGE
        freopen("handle.in", "r", stdin);
        freopen("handle.out", "w", stdout);
    #endif
    
    scanf("%d", &n);
    for (int i = 0; i <= n; i ++)
        scanf("%d", A + i);
    Prepare();
    for (len = (n + 1) << 1; len != (len & -len); len += (len & -len)) ;
    for (int i = len; i > 1; i >>= 1, d ++) ;
    for (int i = 0; i <= n; i ++)
    {
        X[i] = (LL) A[n - i] * Fac[n - i] % Mod * ((i & 1) ? Mod - 1 : 1) % Mod;
        Y[i] = Inv[i];
    }
    for (int i = 0, w = power(g, (Mod - 1) / len); i < len; i ++)
    {
        for (int j = 0; j < d; j ++)
            if ((i >> j) & 1) Rev[i] += 1 << (d - j - 1);
        e[i] = !i ? 1 : (LL) e[i - 1] * w % Mod;
    }
    FFT(X), FFT(Y);
    for (int i = 0, Inv_w = power(power(g, (Mod - 1) / len), Mod - 2); i < len; i ++)
    {
        X[i] = (LL) X[i] * Y[i] % Mod;
        e[i] = !i ? 1 : (LL) e[i - 1] * Inv_w % Mod;
    }
    FFT(X);
    for (int i = 0, Inv_len = power(len, Mod - 2); i < len; i ++)
        X[i] = (LL) X[i] * Inv_len % Mod * Inv[n - i] % Mod * ((i & 1) ? Mod - 1 : 1) % Mod;
    for (int i = n; ~i; i --)
        printf("%d%c", X[i], !i ? '
' : ' ');
    
    #ifndef ONLINE_JUDGE
        fclose(stdin);
        fclose(stdout);
    #endif
    return 0;
}