洛谷 P4827 [国家集训队] Crash 的文明世界（第二类斯特林数 + 树形DP + 思维）

---

用第二类斯特林公式代入$dist(i,j{)}^k$（接下来用$d_j$表示dis(i,j)）：$$S(i)=\sum _{j=1}^n{d}_j^k$$$$=\sum _{j=1}^n\sum _{t=0}^{k}S(k,t)[d_j{]}_t$$$$=\sum _{j=1}^n\sum _{t=0}^{k}S(k,t)C(d_j,t)\ast t!$$$$=\sum _{t=0}^{k}S(k,t)\ast t!\sum _{j=1}^{n}C(d_j,t)$$

变成要快速求解${\sum }_{j=1}^{n}C(d_j,t)$，接下来一步比较巧妙。
由于暴力是必定要T的，应该考虑是否能从其它点的答案转移以缩小时间复杂度。

对于二项式有个递推式：${\sum }_{j=1}^{n}C(d_j,t)={\sum }_{j=1}^{n}C(d_j-1,t)+{\sum }_{j=1}^{n}C(d_j-1,t-1)$

满足$d_j-1$的点一定是当前点的子结点或父节点，这意味着可以树形DP，用两次扫描换根来转移。
令$dp[i][t]$表示结点 $i$ 的${\sum }_{j=1}^{n}C(d_j,t)$,$dp[u][t]=dp[v][t]+dp[v][t-1],v=son(u)|father(u)$
复杂度为$O(nk)$

---

代码：

#include<bits/stdc++.h>
using namespace std;
const int maxn = 5e4 + 100;
const int mod = 10007;
int n,k;
vector<int> g[maxn];
int S[300][300],dp[maxn][300],pw[maxn];
int cur[300];
void dfs(int u,int f) {
	dp[u][0] = 1;
	for(auto it : g[u]) {
		if(it == f) continue;
		dfs(it,u);
		dp[u][0] += dp[it][0];
		dp[u][0] %= mod;
		for(int j = 1; j <= k; j++) {
			dp[u][j] += (dp[it][j - 1] + dp[it][j]) % mod;
			dp[u][j] %= mod;
		}
	}
}
void dfs2(int u,int f) {
	for(auto it : g[u]) {
		if(it == f) continue;
		cur[0] = (dp[u][0] - dp[it][0] + mod) % mod;
		for(int j = 1; j <= k; j++) {
			cur[j] = dp[u][j] - dp[it][j] - dp[it][j - 1];
			cur[j] = (cur[j] + mod + mod) % mod;
		}
		dp[it][0] = (dp[it][0] + cur[0]) % mod;
		for(int j = 1; j <= k; j++) {
			dp[it][j] += (cur[j] + cur[j - 1]) % mod;
			dp[it][j] %= mod;
		}
		dfs2(it,u);
	}
}
int main() {
	scanf("%d%d",&n,&k);
	for(int i = 1; i < n; i++) {
		int u,v;scanf("%d%d",&u,&v);
		g[u].push_back(v);
		g[v].push_back(u);
	}
	pw[0] = S[0][0] = 1;
	for(int i = 1; i <= 300; i++) {
		pw[i] = pw[i - 1] * i % mod;
		for(int j = 1; j <= i; j++)
			S[i][j] = (S[i - 1][j - 1] + j * S[i - 1][j] % mod) % mod;
	}
	dfs(1,0);dfs2(1,0);
	/*for(int i = 1; i <= n; i++)
		for(int j = 1; j <= k; j++)
			printf("**%d\n",dp[i][j]);*/
	int ans = 0;
	for(int i = 1; i <= n; i++) {
		ans = 0;
		for(int j = 0; j <= k; j++) {
			ans += dp[i][j] * S[k][j] % mod * pw[j] % mod;
			ans %= mod;
		}
		printf("%d\n",ans);
	}
}