51NOD 序列求和 V5题解-CSDN博客

本文链接：https://blog.csdn.net/qkhhsuhrtfk/article/details/53428527

本文详细解析了51NOD序列求和V5题目，涉及到数论概念和完全积性函数的线性筛预处理。当R=0时，答案为0；R=1时，通过二项式反演求解；R≠1时，利用特定的多项式关系式进行计算。文章还证明了一个关于求和的引理，有助于解题。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

　　被数论くん搞得头大的本宝宝来写一发数论题~
　　51NOD 序列求和 V5
　　若 $R \equiv 0$ ，那么显然答案就为0.
　　接下来，我们用符号 $f_{k}(n)$ 代表我们所要求的值。首先我们要预处理 $f_{k}(1),\cdots,f_{k}(k + 1)$ 的值，由于 $x^{k}$ 是一个完全积性函数，利用线性筛，这一预处理可以很容易在 $O(k)$ 的时间内完成。
　　若 $R \equiv 1$ ，则原问题转化为求 $\sum_{i = 1}^{n} i ^{k}$ 的值，我们知道，这时 $f_{k}(n)$ 是关于 $n$ 的 $k +1$ 次多项式。同时，我们也知道， $n \ge k +1$ 时，每一个 $C_{n}^{i}$ 都是关于 $n$ 的 $i$ 次多项式，并且它们线性无关，而 $n \leq k$ 时，可以 $O(1)$ 直接查询。对于 $n \ge k +1$ 的情况，我们用 $C_{n}^{0}, \cdots, C_{n}^{k +1}$ 这 $k +2$ 个数来线性表出 $f_{k}(n)$ ，设 $f_{k}(n) = \sum_{i = 0}^{k + 1}C_{n}^{i}a_{i}$ ，根据二项式反演，有 $a_{n} = \sum_{i = 0}^{n}(-1)^{n - i}C_{n}^{i}f_{k}(i)$ 。
　　这里先写一个引理： $\sum_{i = 0}^{k}(-1)^{i}C_{n}^{i} = (-1)^{k}C_{n - 1}^{k}$ ，这个很好证我就不写了。
　　将 $a_{n}$ 代回原式得：

f k (n) = \sum i = 0 k + 1 C i n \sum j = 0 i (- 1) i - j C j i f k (j) = \sum i = 0 k + 1 f k (i) \sum j = i k + 1 (- 1) j - i C j n C i j = \sum i = 0 k + 1 f k (i) C i n \sum j = i k + 1 (- 1) j - i C j - i n - i = \sum i = 0 k + 1 f k (i) C i n \sum j = 0 k - i + 1 (- 1) j C j n - i = \sum i = 0 k + 1 (- 1) k - i + 1 C i n C k - i + 1 n - i - 1 = \sum i = 0 k + 1 \prod 0 \leq j \leq k + 1 , j \neq i n - j i ! ( k - i + 1 ) !

$\begin {aligned} f_{k}(n) &=\sum_{i = 0}^{k +1}C_{n}^{i}\sum_{j = 0}^{i}(-1)^{i - j}C_{i}^{j}f_{k}(j)\\ &=\sum_{i = 0}^{k + 1}f_{k}(i)\sum_{j = i}^{k +1}(-1)^{j - i}C_{n}^{j}C_{j}^{i}\\ &=\sum_{i = 0}^{k + 1}f_{k}(i)C_{n}^{i}\sum_{j = i}^{k + 1}(-1)^{j - i}C_{n - i}^{j - i}\\ &=\sum_{i = 0}^{k + 1}f_{k}(i)C_{n}^{i}\sum_{j = 0}^{k - i + 1}(-1)^{j}C_{n - i}^{j}\\ &=\sum_{i = 0}^{k + 1}(-1)^{k - i + 1}C_{n}^{i}C_{n - i - 1}^{k - i + 1}\\ &=\sum_{i = 0}^{k + 1}\frac{\prod_{0 \leq j \leq k + 1, j \neq i}n - j}{i!(k - i + 1)!} \end {aligned}\\$

　　这样，分子和分母的部分都可以预处理出来，复杂度 $O(k)$ 。
　　若 $R \not\equiv 1$ ，有：

(R - 1) f k (n) ∴ f k (n) = (R - 1) \sum i = 1 n i k R i = \sum i = 1 n i k R i + 1 - \sum i = 1 n i k R i = \sum i = 1 n + 1 (i - 1) k R i - \sum i = 1 n i k R i = n k R n + 1 - \sum i = 1 n \sum j = 0 k - 1 (- 1) k - j - 1 C j k i j R i = n k R n + 1 - \sum j = 0 k - 1 (- 1) k - j + 1 C j k f j (n) = n k R n + 1 - \sum k - 1 j = 0 ( - 1 ) k - j + 1 C j k f j ( n ) R - 1

$\begin{aligned} (R - 1)f_{k}(n) &= (R - 1)\sum_{i = 1}^{n}i^{k}R^{i}\\ &=\sum_{i = 1}^{n}i^{k}R^{i + 1} - \sum_{i = 1}^{n}i^{k}R^{i}\\ &=\sum_{i = 1}^{n + 1}(i - 1)^{k}R^{i} - \sum_{i = 1}^{n}i^{k}R^{i}\\ &=n^{k}R^{n + 1}-\sum_{i = 1}^{n}\sum_{j = 0}^{k - 1}(-1)^{k - j - 1}C_{k}^{j}i^{j}R^{i}\\ &=n^{k}R^{n + 1} - \sum_{j = 0}^{k - 1}(-1)^{k - j + 1}C_{k}^{j}f_{j}(n)\\ \therefore f_{k}(n) &= \frac{n^{k}R^{n + 1} - \sum_{j = 0}^{k - 1}(-1)^{k - j + 1}C_{k}^{j}f_{j}(n)}{R - 1}\\ \end{aligned}\\$

　　下面我们证明一个结论，对 $\forall f_{i}(n), 0 \leq i \leq k, \exists F_{i}(n)\in Q^{i}[x]$ ，有 $f_{i}(n) = R^{n + 1}F_{i}(n) - RF_{i}(0)$ 。

证 明 ： 当 i 当 i f k (n) 这 样 ， = 0 时, 结 论 已 经 成 立 (请 自 行 验 证) = k 时 ， 假 设 结 论 对 0, \dots, k - 1 均 成 立 = n k R n + 1 - \sum k - 1 j = 0 ( - 1 ) k - j + 1 C j k f j ( n ) R - 1 = n k R n + 1 - \sum k - 1 j = 0 ( - 1 ) k - j + 1 C j k [ R n + 1 F j ( n ) - R F j ( 0 ) ] R - 1 = R n + 1 [ n k - \sum k - 1 j = 0 ( - 1 ) k - j + 1 C k j F j ( n ) ] + R [ \sum k - 1 j = 0 ( - 1 ) k - j + 1 C j k F j ( 0 ) ] R - 1 我 们 令 F k (n) = n k - \sum j = 0 k - 1 (- 1) k - j + 1 C k j F j (n) ， 容 易 验 证 它 满 足 题 目 要 求 。

$\begin{aligned}\\ 证明：当i &= 0时,结论已经成立(请自行验证)\\ 当i &= k时，假设结论对0,\cdots,k - 1均成立\\ f_{k}(n) &= \frac{n^{k}R^{n + 1} - \sum_{j = 0}^{k - 1}(-1)^{k - j + 1}C_{k}^{j}f_{j}(n)}{R - 1}\\ &=\frac{n^{k}R^{n + 1}-\sum_{j = 0}^{k - 1}(-1)^{k - j + 1}C_{k}^{j}[R^{n + 1}F_{j}(n)-RF_{j}(0)]}{R - 1}\\ &=\frac{R^{n + 1}[n^{k}-\sum_{j = 0}^{k - 1}(-1)^{k - j + 1}C_{j}^{k}F_{j}(n)]+R[\sum_{j = 0}^{k - 1}(-1)^{k - j + 1}C_{k}^{j}F_{j}(0)]}{R - 1}\\ 这样，&我们令F_{k}(n) = n^{k}-\sum_{j = 0}^{k - 1}(-1)^{k - j + 1}C_{j}^{k}F_{j}(n)，容易验证它满足题目要求。 \end{aligned}\\$
　　我们再来证明一个引理：对

∀0≤m≤k,∑ k i=0 (−1) i C i k C m i =0 $\forall 0 \leq m \leq k, \sum_{i = 0}^{k}(-1)^{i}C_{k}^{i}C_{i}^{m} = 0$

证 明 ： \sum i = 0 k (- 1) i C i k C m i = \sum i = 0 k (- 1) i C m k C i - m k - m = C m k \sum i = 0 k - m (- 1) i + m C i k - m = C m k (- 1) m (1 - 1) (k - m) = 0

$\begin{aligned}\\ 证明：\sum_{i = 0}^{k}(-1)^{i}C_{k}^{i}C_{i}^{m} &= \sum_{i = 0}^{k}(-1)^{i}C_{k}^{m}C_{k - m}^{i - m}\\ &=C_{k}^{m}\sum_{i = 0}^{k - m}(-1)^{i + m}C_{k - m}^{i}\\ &=C_{k}^{m}(-1)^{m}(1 - 1)^{(k - m)}\\ &=0\\ \end{aligned}\\$
　　在上一部分的讨论中，我们已经知道，作为一个

k $k$ 次多项式，

F k (n) $F_{k}(n)$ 可以表示为

C 0 n ,⋯,C k n $C_{n}^{0},\cdots,C_{n}^{k}$ 的线性组合。设

F k (n)=∑ k i=0 C i n a i $F_{k}(n) = \sum_{i = 0}^{k}C_{n}^{i}a_{i}$ ，则

\sum i = 0 k (- 1) i C i k F k (n) = \sum i = 0 k (- 1) i C i k \sum j = 0 k C j i a j = \sum j = 0 k \sum i = 0 k (- 1) i C i k C j i a j = \sum j = 0 k 0 = 0

$\begin{aligned}\\ \sum_{i = 0}^{k}(-1)^{i}C_{k}^{i}F_{k}(n) &= \sum_{i = 0}^{k}(-1)^{i}C_{k}^{i}\sum_{j = 0}^{k}C_{i}^{j}a_{j}\\ &= \sum_{j = 0}^{k}\sum_{i = 0}^{k}(-1)^{i}C_{k}^{i}C_{i}^{j}a_{j}\\ &=\sum_{j = 0}^{k}0\\ &= 0\\ \end{aligned}\\$
　　我们又有：

(n + 1) k R n + 1 ∴ F k (n + 1) = f k (n + 1) - f k (n) = R n + 2 F k (n + 1) - R n + 1 + R F k (0) = ( n + 1 ) k + F k ( n ) R

$\begin{aligned}\\ (n + 1)^{k}R^{n + 1} &= f_k(n + 1) - f_k(n)\\ &=R^{n + 2}F_k(n + 1) - R^{n + 1} + RF_k(0)\\ \therefore F_k(n + 1) &= \frac{(n + 1)^{k} + F_k(n)}{R} \end{aligned}\\$
　　通过上面两个等式，我们就可以通过解一个一元一次方程来得到

F k (0),⋯,F k (k) $F_k(0),\cdots,F_k(k)$ 的值，之后我们用拉格朗日插值公式就可以

O(k) $O(k)$ 地得到

F k (n) $F_k(n)$ 的值，再代回原式，这道题就解决了。

#include<bits/stdc++.h>
#define ll long long

const int N = 200010;
const int moder = 985661441;

int power(int a, ll index){
    int ans = 1;
    while (index){
        if (index & 1)
            ans = 1ll * ans * a % moder;
        a = 1ll * a * a % moder;
        index >>= 1;
    }
    return ans;
}

int min[N], x[N], f[N], F[N], prefix[N], suffix[N], fac[N], inv[N], t, k;
ll n, r;
std::vector <int> prime;

void calc(){
    prefix[0] = suffix[k + 1] = 1;
    for (int i = 1; i <= k + 1; ++ i){
        prefix[i] = 1ll * prefix[i - 1] * ((n - i + 1) % moder) % moder;
    }
    for (int i = k; i >= 0; -- i){
        suffix[i] = 1ll * suffix[i + 1] * ((n - i - 1) % moder) % moder;
    }
    int ans = 0;
    for(int i = 0; i <= k + 1; ++ i){
        int now = 1ll * f[i] * inv[i] % moder * inv[k - i + 1] % moder;
        now = 1ll * now * prefix[i] % moder * suffix[i] % moder;
        if ((k - i + 1) & 1)
            now = (moder - now) % moder;
        ans = (ans + now) % moder;
    }
    printf("%d\n", ans);
}

void _calc(){
    int inv_r = power(r, moder - 2);
    int sum_a = 1, sum_b = 0;
    prefix[0] = 1;
    suffix[0] = 0;
    for (int i = 1; i <= k + 1; ++ i){
        prefix[i] = 1ll * prefix[i - 1] * inv_r % moder;
        suffix[i] = 1ll * (suffix[i - 1] + x[i]) * inv_r % moder;
        int coe = 1ll * fac[k + 1] % moder * inv[i] % moder * inv[k + 1 - i] % moder;
        if (i & 1)
            coe = (moder - coe) % moder;
        sum_a = (sum_a + 1ll * coe * prefix[i]) % moder;
        sum_b = (sum_b + 1ll * coe * suffix[i]) % moder;
    }
    int _x = 1ll * (moder - sum_b) * power(sum_a, moder - 2) % moder;
    for (int i = 0; i <= k; ++ i)
        F[i] = (1ll * prefix[i] * _x + suffix[i]) % moder;
    prefix[0] = suffix[k] = 1;
    for (int i = 1; i <= k; ++ i){
        prefix[i] = 1ll * prefix[i - 1] * ((n - i + 1) % moder) % moder;
    }
    for (int i = k - 1; i >= 0; -- i){
        suffix[i] = 1ll * suffix[i + 1] * ((n - i - 1) % moder) % moder;
    }
    int ans = 0;
    for(int i = 0; i <= k; ++ i){
        int now = 1ll * F[i] * inv[k - i] % moder * inv[i] % moder;
        now = 1ll * now * prefix[i] % moder * suffix[i] % moder;
        if ((k - i) & 1)
            now = (moder - now) % moder;
        ans = (ans + now) % moder;
    }
    ans = (1ll * ans * power(r, n) - F[0] + moder) % moder;
    ans = 1ll * ans * r % moder;
    printf("%d\n", ans);
}

int main(){
    scanf("%d", &t);
    fac[0] = fac[1] = inv[0] = inv[1] = 1;
    for (int i = 2; i < N; ++ i){
        fac[i] = 1ll * i * fac[i - 1] % moder;
        inv[i] = power(fac[i], moder - 2);
        if (!min[i]){
            min[i] = i;
            prime.push_back(i);
        }
        for (int j = 0; j < prime.size() && prime[j] <= i && prime[j] * i < N; ++ j)
            min[prime[j] * i] = prime[j];
    }
    while (t --){   
        scanf("%d%I64d%I64d", &k, &n, &r);
        r %= moder;
        if (r == 0){
            puts("0\n");
            continue;
        }
        x[1] = 1;
        int now = r;
        f[1] = r;
        for (int i = 2; i < k + 10; ++ i){
            now = 1ll * now * r % moder;
            if (min[i] == i)
                x[i] = power(i, k);
            else
                x[i] = 1ll * x[min[i]] * x[i / min[i]] % moder;
            f[i] = (f[i - 1] + 1ll * x[i] * now) % moder;
        }
        if (n < 1ll * (k + 10)){
            printf("%d\n", f[n]);
            continue;
        }
        if (r == 1){
            calc();
            continue;
        }
        _calc();
    }
    return 0;
}