题解 hdu4625 JZPTREE【第二类斯特林数 树形DP】

题目链接：hdu4625（vjudge）

题意

给定一棵树，对于每个节点 $i$ 求其他所有点到它距离的 $k$ 次方和，答案模 10007。$n\le 5\times 10^4,k\le 500$。

题解

先由儿子到父亲 DP 一遍、再由父亲到儿子 DP 一遍，中途想办法快速维护 $x^k\to (x+1{)}^k$

记 $S(n,m)$ 为第二类斯特林数，即 $n$ 个不同的球放入 $m$ 个相同盒子（不允许空）的方案数。可以预先 $O(m^2)$ 递推。

$$x^k=\sum _{i=0}^k\left(\genfrac{}{}{0ex}{}{k}{i}\right)\cdot S(k,i)\cdot i!=\sum _{i=0}^k{x}^{\underline{i}}S(k,i)$$

其组合意义为：把 $x$ 个不同的球放入 $k$ 个不同盒子里（允许空）的方案数 等于 枚举有球的盒子的数量 再选出这些盒子 再把 $x$ 个球放进这些相同的盒子 再给这些盒子编号。

对于 $0\le i\le m$ 维护 $x^{\underline{i}}$：
$$\begin{array}{rl}& (x+1{)}^{\underline{i}}\\ =& x^{\underline{i-1}}\cdot (x+1)\\ =& (x-i+1)x^{\underline{i-1}}+i\cdot x^{\underline{i-1}}\\ =& x^{\underline{i}}+i\cdot x^{\underline{i-1}}\end{array}$$

最后 $O(m)$ 还原成 $x^m$。

总复杂度 $O(m^2+mn)$。

代码：

/**********
Author: WLBKR5
Problem: hdu 4625
Name: JZPTREE
Source: 2013 Multi-University Training Contest 3
Algorithm: 树形DP, 第二类斯特林数 
Date: 2020/06/09
Statue: accepted
Submission: vjudge.net/solution/25960864
**********/
#include<bits/stdc++.h>
using namespace std;
int getint(){
	int ans=0,f=1;
	char c=getchar();
	while(c<'0'||c>'9'){
		if(c=='-')f=-1;
		c=getchar();
	}
	while(c>='0'&&c<='9'){
		ans=ans*10+c-'0';
		c=getchar();
	}
	return ans;
}
const int M=510,N=5e4+10,mod=10007;
struct bian{
	int e,n;
};
bian b[N<<1];
int s[N],tot=0;
void add(int x,int y){
	tot++;
	b[tot].e=y;
	b[tot].n=s[x];
	s[x]=tot;
}
int n,m;

int st[M][M];
struct P{
	int a[M];
	P(){ a[0]=1; }
	P _() const {
		P ans;
		ans.a[0]=a[0];
		for(int i=1;i<=m;i++)ans.a[i]=(a[i]+a[i-1]*1ll*i)%mod;
		return ans;
	}
	void operator+= (const P& x){
		for(int i=0;i<=m;i++){
			a[i]+=x.a[i];
			if(a[i]>=mod)a[i]-=mod;
		}
	}
	const P operator- (const P& x){
		P ans;
		for(int i=0;i<=m;i++){
			ans.a[i]=a[i]-x.a[i];
			if(ans.a[i]<0)ans.a[i]+=mod; 
		}
		return ans;
	}
	operator int(){
		int ans=0;
		for(int i=0;i<=m;i++)ans=(ans+st[m][i]*1ll*a[i])%mod;
		return ans; 
	}
};

P f[N],g[N];
void ss1(int x,int fa){
	for(int i=s[x];i;i=b[i].n){
		if(b[i].e==fa)continue;
		ss1(b[i].e,x);
		f[x]+=f[b[i].e]._();
	}
}
void ss2(int x,int fa){
	for(int i=s[x];i;i=b[i].n){
		if(b[i].e==fa)continue;
		f[b[i].e]+=(f[x]-f[b[i].e]._())._();
		ss2(b[i].e,x);
	}
}

int main(){
	for(int i=0;i<=500;i++)st[0][i]=st[i][0]=!i;
	for(int i=1;i<=500;i++){
		for(int j=1;j<=500;j++){
			st[i][j]=(st[i-1][j-1]+st[i-1][j]*1ll*j)%mod;
			//cerr<<"> "<<i<<" "<<j<<" "<<st[i][j]<<endl;
		}
	}
	int T=getint();
	while(T--){
		n=getint(),m=getint();
		memset(s,0,sizeof(s));tot=0;
		memset(f,0,sizeof(f));memset(g,0,sizeof(g));
		for(int i=1;i<=n;i++)f[i].a[0]=g[i].a[0]=1;
		for(int i=1;i<n;i++){
			int x=getint(),y=getint();
			add(x,y);
			add(y,x);
		}
		ss1(1,0);
		ss2(1,0);
		for(int i=1;i<=n;i++){
			printf("%d\n",int(f[i]));
		}
	}
	return 0;
}