【hdu 7085】Pty loves SegmentTree (推式子)

题目链接

题目大意

一棵线段树，分割点任选，每个节点有1个权值。规定：

• 叶节点权值为1
• 右子树线段长度恰好为 $k$ 的节点权值为 $A$
• 其他节点权值为 $B$

定义线段树的权值为所有节点权值之积。
定义 $f_n$ 为区间 $[1,n]$ 所有可能的线段树的权值和。
给定 $k,A,B$，多组询问，每次问 ${\sum }_{i=L}^R{f}_i^2$ 的值（$L,R\le 10^7$）。

思路

其实就是要在 $O(n)$ 的时间内求出 $f_1\sim f_n$ 的值。
首先不难写出递推式
$$f_n=(A-B)f_k{f}_{n-k}+B\sum _{i=1}^{n-1}{f}_i{f}_{n-i}(i\ge 2)$$
$$f_1=1$$
定义
$$F(x)=\sum _{i=1}^{\infty }{f}_i{x}^i$$
则
$$F=(A-B)f_k{x}^{k}F+B{F}^2+x$$
解得
$$F=\frac{-((A-B)f_k{x}^k-1)-\sqrt{((A-B)f_k{x}^k-1{)}^2-4Bx}}{2B}$$

为简化式子，可以先计算出
$$f_k=[x^k]F(x)=[x^k]\frac{1-\sqrt{1-4Bx}}{2B}=\frac{(2k-3)!!(2B{)}^{k-1}}{k!}=\frac{1}{2k-1}\left(\genfrac{}{}{0ex}{}{2k-1}{k}\right)B^{k-1}$$
令 $C=(A-B)f_k$
则
$$F=\frac{-(C{x}^k-1)-\sqrt{(C{x}^k-1{)}^2-4Bx}}{2B}$$

但那个根式有点麻烦，因此我们需要尝试进一步化简。
令
$$Q=\sqrt{(C{x}^k-1{)}^2-4Bx}$$
则
$$Q=1-C{x}^k-2BF$$
求导得
$$Q^{\prime }=-kC{x}^{k-1}-2B{F}^{\prime }$$
而对 $Q^2$ 求导得
$$Q{Q}^{\prime }=k{C}^2{x}^{2k-1}-kC{x}^{k-1}-2B$$
带入 $Q^2{Q}^{\prime }=(Q{Q}^{\prime })Q$得
$$(C^2{x}^{2k}-2C{x}^k+1-4Bx)(kC{x}^{k-1}+2B{F}^{\prime })+(k{C}^2{x}^{2k-1}-kC{x}^{k-1}-2B)(1-C{x}^k-2BF)=0$$
即得到关于 $F$ 和 $F^{\prime }$ 的一次式。
整理得
$$(C^2{x}^{2k}-2C{x}^k+1-4Bx)F^{\prime }-(k{C}^2{x}^{2k-1}-kC{x}^{k-1}-2B)F-(2k-1)C{x}^k-1=0$$
取 $n$ 次项，得
$$(n-3k+1)C^2{f}_{n-2k+1}-(2n-3k+2)C{f}_{n-k+1}+(-4Bn+2B)f_n+(n+1)f_{n+1}-(2k-1)C[n==k]=0$$
即
$$f_{n+1}=\frac{1}{n+1}(2B(2n-1)f_n+(2n-3k+2)C{f}_{n-k+1}-(n-3k+1)C^2{f}_{n-2k+1}+(2k-1)C[n==k])$$
即
$$f_i=\frac{1}{i}(2B(2i-3)f_{i-1}+(2i-3k)C{f}_{i-k}-(i-3k)C^2{f}_{i-2k}+(2k-1)C[i==k+1])$$

代码

#include <bits/stdc++.h>
#define rep(i, l, r) for (int i = l; i <= r; ++i)
#define per(i, r, l) for (int i = r; i >= l; --i)
using namespace std;
const int mod = 998244353;
const int N = 20000007;
int fac[N], invfac[N];
int _inv[N];
int pw(int x, int y) {
    int ret = 1;
    while (y) {
        if (y & 1) ret = 1ll * ret * x % mod;
        x = 1ll * x * x % mod;
        y >>= 1;
    }
    return ret;
}
int inv(int x) { return pw(x, mod - 2); }
void init() {
    int lim = 20000000;
    fac[0] = 1;
    rep(i, 1, lim) { fac[i] = 1ll * fac[i - 1] * i % mod; }
    invfac[lim] = inv(fac[lim]);
    per(i, lim - 1, 0) { invfac[i] = 1ll * invfac[i + 1] * (i + 1) % mod; }
    _inv[1] = 1;
    rep(i, 2, lim) { _inv[i] = 1ll * (mod - mod / i) * _inv[mod % i] % mod; }
}
int Comb(int n, int k) {
    return 1ll * fac[n] * invfac[k] % mod * invfac[n - k] % mod;
}

int T;
int Q, k, A, B;
int f[N], s[N];
void work() {
    f[k] = 1ll * Comb(2 * k - 1, k - 1) * pw(B, k - 1) % mod * _inv[2 * k - 1] %
           mod;
    int C = 1ll * f[k] * (A - B + mod) % mod;
    // printf("C=%d\n", C);
    f[1] = 1;
    int lim = 10000000;
    rep(i, 2, lim) {
        int ret = 2ll * B * (2 * i - 3) % mod * f[i - 1] % mod;
        if (i > k) {
            ret =
                (ret + 1ll * (2 * i - 3 * k + mod) * C % mod * f[i - k]) % mod;
        }
        if (i > 2 * k) {
            ret = (ret +
                   1ll * (3 * k - i + mod) * C % mod * C % mod * f[i - 2 * k]) %
                  mod;
        }
        if (i == k + 1) {
            ret = (ret + 1ll * (2 * k - 1) * C) % mod;
        }
        ret = 1ll * ret * _inv[i] % mod;
        f[i] = ret;
        // if (i <= 100) printf("f[%d]=%d\n", i, f[i]);
    }
    rep(i, 1, lim) { s[i] = (s[i - 1] + 1ll * f[i] * f[i]) % mod; }
}
int solve(int l, int r) { return (s[r] - s[l - 1] + mod) % mod; }
int main() {
    init();
    scanf("%d", &T);
    while (T--) {
        scanf("%d%d%d%d", &Q, &k, &A, &B);
        work();
        while (Q--) {
            int l, r;
            scanf("%d%d", &l, &r);
            printf("%d\n", solve(l, r));
        }
    }
    return 0;
}