【BZOJ3157】【BZOJ3516】【51nod1229】国王奇遇记 & 数列求和V2（自然数幂和，递推）

Description

求$\displaystyle \sum _{i=1}^n i^k r^i$

Solution

当$r=1$时，就是个裸的自然数幂和问题。

如果$r\not = 1$，类似于求自然数幂和时的递推做法，设$S(k)=\sum\limits_{i=1}^n i^k r^i$
那么$\displaystyle r S(k)=\sum_{i=1}^ni^kr^{i+1}=\sum_{i=1}^{n+1}(i-1)^kr^{i+1}$
所以有：

$$\begin{align*} (r-1)S(k) &= n^kr^{n+1}+\sum_{i=1}^n[(i-1)^k-i^k]r^i \\ &= n^kr^{n+1}+\sum_{i=1}^n\sum_{j=0}^{k-1}(-1)^{k-j}\binom{k}{j}i^jr^i \\ &= n^kr^{n+1}+\sum_{j=0}^{k-1}(-1)^{k-j}\binom{k}{j}\sum_{i=1}^n i^jr^i\\ &= n^kr^{n+1}+\sum_{j=0}^{k-1}(-1)^{k-j}\binom{k}{j}S(j) \end{align*}$$
直接递推即可。

---

Code

inline LL Pow(LL a, LL b)
{
    LL Ans = 1;
    for (a %= Mod ; b; b >>= 1, (a *= a) %= Mod) if (b & 1) (Ans *= a) %= Mod;
    return Ans;
}

inline LL C(int n, int m) { return fac[n] * ifac[m] % Mod * ifac[n - m] % Mod; }

LL S1(int n, int k)
{
    if (f[k]) return f[k];
    f[k] = pown[k + 1] - 1;
    rep(i, k) (f[k] -= C(k + 1, i) * S1(n, i) % Mod) %= Mod;
    ((((f[k] *= Pow(k + 1, Mod - 2)) %= Mod) += Mod) %= Mod);
    return f[k];
}

int main()
{
#ifdef hany01
    File("51nod1229");
#endif

    fac[0] = 1;
    For(i, 1, maxk - 5) fac[i] = fac[i - 1] * i % Mod;
    ifac[maxk - 5] = Pow(fac[maxk - 5], Mod - 2);
    Fordown(i, maxk - 5, 1) ifac[i - 1] = ifac[i] * i % Mod;

    for (static int T = read(); T --; ) {
        scanf("%lld", &n), k = read(), scanf("%lld", &r);

        if (r == 1) {
            pown[0] = 1;
            For(i, 1, k + 1) pown[i] = pown[i - 1] * (n + 1) % Mod;
            printf("%lld\n", S1(n, k));
            continue;
        }

        LL inv = Pow(r - 1, Mod - 2);
        S[0] = (Pow(r, n + 1) - r) % Mod * inv % Mod;
        For(i, 1, k) {
            S[i] = Pow(n, i) * Pow(r, n + 1) % Mod;
            For(j, 0, i - 1) (S[i] += (((i - j) & 1 ? -1 : 1) * C(i, j) * S[j] % Mod)) %= Mod;
            (S[i] *= inv) %= Mod;
        }
        printf("%lld\n", (S[k] + Mod) % Mod);
    }

    return 0;
}
//共看明月应垂泪，一夜乡心五处同。
//    -- 白居易《望月有感》