[国家集训队]Crash的文明世界 题解

传送门

题意

给一棵树，设$S(x)=\sum _{i=1}^n\mathrm{d}\mathrm{i}\mathrm{s}\mathrm{t}(x,i{)}^k$，对每个点求出$S(x)$。

$n\le 5\times 10^4,k\le 150$。

题解

不会点分树

首先我们知道第二类斯特林数$S_i^j$满足这样一个关系

$$m^n=\sum _{i=0}^n{S}_n^{i}i!C_m^i$$

套进去

$$S(x)=\sum _{i=1}^n\sum _{j=0}^k{S}_k^{j}j!C_{\mathrm{d}\mathrm{i}\mathrm{s}\mathrm{t}(i,x)}^j$$

交换求和符号

$$S(x)=\sum _{j=0}^k{S}_k^{j}j!\sum _{i=1}^n{C}_{\mathrm{d}\mathrm{i}\mathrm{s}\mathrm{t}(i,x)}^j$$
于是我们只要求后面那个$\sum _{i=1}^n{C}_{\mathrm{d}\mathrm{i}\mathrm{s}\mathrm{t}(i,x)}^j$即可。

树形dp，设$f(x,j)=\sum _{u\in X}{C}_{\mathrm{d}\mathrm{i}\mathrm{s}\mathrm{t}(x,u)}^j$，其中$X$表示以$x$为根的子树（包括$x$本身）。
$$f(x,j)=\sum _{u\in X}(C_{\mathrm{d}\mathrm{i}\mathrm{s}\mathrm{t}(x,u)-1}^j+C_{\mathrm{d}\mathrm{i}\mathrm{s}\mathrm{t}(x,u)-1}^{j-1})$$

$u$到$x$的距离为$\mathrm{d}\mathrm{i}\mathrm{s}\mathrm{t}(x,u)$，$u$又是$X$中的点，那么与$u$相距$\mathrm{d}\mathrm{i}\mathrm{s}\mathrm{t}(x,u)-1$的点也就是$x$的孩子了。所以我们有转移

$$f(x,j)=\sum _{v\in \mathrm{s}\mathrm{o}\mathrm{n}(x)}f(v,j)+f(v,j-1)$$

其中$\mathrm{s}\mathrm{o}\mathrm{n}(x)$是$x$的孩子集合。

这样还不够，我们还需要求$g(x,j)=\sum _{u\notin X}{C}_{\mathrm{d}\mathrm{i}\mathrm{s}\mathrm{t}(x,u)}^j$。

$$g(x,j)=\sum _{u\notin X}(C_{\mathrm{d}\mathrm{i}\mathrm{s}\mathrm{t}(x,u)-1}^j+C_{\mathrm{d}\mathrm{i}\mathrm{s}\mathrm{t}(x,u)-1}^{j-1})$$

$$=\sum _{u\notin X}(C_{\mathrm{d}\mathrm{i}\mathrm{s}\mathrm{t}(\mathrm{f}\mathrm{a}(x),u)}^j+C_{\mathrm{d}\mathrm{i}\mathrm{s}\mathrm{t}(\mathrm{f}\mathrm{a}(x),u)}^{j-1})$$
其中
$$\sum _{u\notin X}{C}_{\mathrm{d}\mathrm{i}\mathrm{s}\mathrm{t}(\mathrm{f}\mathrm{a}(x),u)}^j=f(\mathrm{f}\mathrm{a}(x),j)+g(\mathrm{f}\mathrm{a}(x),j)-f(x,j)-f(x,j-1)$$
$$\sum _{u\notin X}{C}_{\mathrm{d}\mathrm{i}\mathrm{s}\mathrm{t}(\mathrm{f}\mathrm{a}(x),u)}^{j-1}=f(\mathrm{f}\mathrm{a}(x),j-1)+g(\mathrm{f}\mathrm{a}(x),j-1)-f(x,j-1)-f(x,j-2)$$
做完了。

至于斯特林数怎么求，由于这里只要$k\times k$范围内的，所以直接利用递推关系

$$S_i^j=S_{i-1}^{j-1}+j{S}_{i-1}^j$$

就行了，不需要NTT之类的复杂玩意。

#include <cctype>
#include <cstdio>
#include <climits>
#include <algorithm>
#include <vector>

template <typename T> inline void read(T& x) {
    int f = 0, c = getchar(); x = 0;
    while (!isdigit(c)) f |= c == '-', c = getchar();
    while (isdigit(c)) x = x * 10 + c - 48, c = getchar();
    if (f) x = -x;
}
template <typename T, typename... Args>
inline void read(T& x, Args&... args) {
    read(x); read(args...); 
}
template <typename T> void write(T x) {
    if (x < 0) x = -x, putchar('-');
    if (x > 9) write(x / 10);
    putchar(x % 10 + 48);
}
template <typename T> inline void writeln(T x) { write(x); puts(""); }
template <typename T> inline bool chkmin(T& x, const T& y) { return y < x ? (x = y, true) : false; }
template <typename T> inline bool chkmax(T& x, const T& y) { return x < y ? (x = y, true) : false; }

const int mod = 10007;
const int maxn = 5e4 + 7, maxk = 157;

int stir[maxk][maxk], f[maxn][maxk], g[maxn][maxk], fac[maxk];
std::vector<int> G[maxn];
int n, K;

void dfs1(int x, int fa) {
    f[x][0] = 1;
    for (auto v : G[x]) if (v != fa) {
        dfs1(v, x);
        f[x][0] = (f[x][0] + f[v][0]) % mod;
        for (int j = 1; j <= K; ++j)
            f[x][j] = (f[x][j] + f[v][j] + f[v][j - 1]) % mod;
    }
}
void dfs2(int x, int fa) {
    if (x > 1) {
        g[x][0] = n - f[x][0];
        g[x][1] = ((f[fa][1] + g[fa][1] + f[fa][0] + g[fa][0] - f[x][1] - 2 * f[x][0]) % mod + mod) % mod;
        for (int j = 2; j <= K; ++j)
            g[x][j] = ((f[fa][j] + g[fa][j] + f[fa][j - 1] + g[fa][j - 1] - f[x][j] - 2 * f[x][j - 1] - f[x][j - 2]) % mod + mod) % mod;
    }
    for (auto v : G[x]) if (v != fa) dfs2(v, x);
}

int main() {
    read(n, K);
    for (int i = 1, x, y; i < n; ++i) {
        read(x, y);
        G[x].push_back(y);
        G[y].push_back(x);
    }
    stir[0][0] = 0; stir[1][1] = 1;
    for (int i = 2; i <= K; ++i)
        for (int j = 1; j <= i; ++j)
            stir[i][j] = (stir[i - 1][j - 1] + j * stir[i - 1][j] % mod) % mod;
    fac[0] = 1;
    for (int i = 1; i <= K; ++i) fac[i] = fac[i - 1] * i % mod;
    dfs1(1, 0);
    dfs2(1, 0);
    for (int i = 1; i <= n; ++i) {
        int si = 0;
        for (int j = 0; j <= K; ++j)
            si = (si + stir[K][j] * fac[j] % mod * (f[i][j] + g[i][j]) % mod) % mod;
        writeln(si);
    }
    return 0;
}