932E-TeamWork 二项式定理+求导/DP

题目链接

932E

题解

题意

求 ${\sum }_{i=0}^n\left(\begin{array}{c}n\\ i\end{array}\right)\ast i^k.$

思路

$$f(x)=(1+x{)}^n=\sum _{i=0}^n\left(\begin{array}{c}n\\ i\end{array}\right)\ast x^i$$
$$x{f}^{\prime }(x)=\sum _{i=0}^n\left(\begin{array}{c}n\\ i\end{array}\right)\ast x^i\ast i$$
$$......$$
$$x^k\ast f^{(k)}(x)=\sum _{i=0}^n\left(\begin{array}{c}n\\ i\end{array}\right)\ast i^k\ast x^i$$
其中，令x == 1即为所求。

我们定义$f[a][b][c]=对x^b\ast (1+x{)}^c求导并乘以x，n次$。则 $f[k][0][n]$ 即为所求。
转移方程为：$f[a][b][c]=b\ast f[a-1][b][c]+c\ast f[a-1][b+1][c-1]$，
其中当 $a==0$ 时，$f[a][b][c]=2^c$。

其中b、c之和恒等于n，故可以将数组降至三维。

AC代码

递归

#include <bits/stdc++.h>
using namespace std;
using ll = long long;
int const N = 5e3 + 10;
int const Mod = 1e9 + 7;
int f[N][N], n, k;

int FastPower(int a, int k) {
    int ans = 1;
    while (k) {
        if (k & 1) ans = 1LL * ans * a % Mod;
        a = 1LL * a * a % Mod;
        k >>= 1;
    }
    return ans;
}

int dp(int a, int b) {
    if (b > n) return 0;
    if (f[a][b] >= 0) return f[a][b];
	if (a == 0) return (f[a][b] = FastPower(2, n - b));
	return (f[a][b] = (1LL * b * dp(a - 1, b) % Mod + 1LL * (n - b) * dp(a - 1, b + 1) % Mod) % Mod);
}

int main() {
    scanf("%d%d", &n, &k);
    memset(f, -1, sizeof(f));
    printf("%d\n", dp(k, 0));
    return 0;
}

参考资料

https://blog.csdn.net/Charlie_jilei/article/details/79330611?utm_medium=distribute.pc_relevant.none-task-blog-BlogCommendFromBaidu-2.channel_param&depth_1-utm_source=distribute.pc_relevant.none-task-blog-BlogCommendFromBaidu-2.channel_param