题意
给定一棵树,对于每个节点 i i i 求其他所有点到它距离的 k k k 次方和,答案模 10007。 n ≤ 5 × 1 0 4 , k ≤ 500 n\leq 5\times 10^4,k\leq 500 n≤5×104,k≤500。
题解
先由儿子到父亲 DP 一遍、再由父亲到儿子 DP 一遍,中途想办法快速维护 x k → ( x + 1 ) k x^k\to (x+1)^k xk→(x+1)k
记 S ( n , m ) S(n,m) S(n,m) 为第二类斯特林数,即 n n n 个不同的球放入 m m m 个相同盒子(不允许空)的方案数。可以预先 O ( m 2 ) O(m^2) O(m2) 递推。
x k = ∑ i = 0 k ( k i ) ⋅ S ( k , i ) ⋅ i ! = ∑ i = 0 k x i ‾ S ( k , i ) x^k=\sum\limits_{i=0}^k{k \choose i}\cdot S(k,i)\cdot i!=\sum\limits_{i=0}^kx^{\underline i}S(k,i) xk=i=0∑k(ik)⋅S(k,i)⋅i!=i=0∑kxiS(k,i)
其组合意义为:把 x x x 个不同的球放入 k k k 个不同盒子里(允许空)的方案数 等于 枚举有球的盒子的数量 再选出这些盒子 再把 x x x 个球放进这些相同的盒子 再给这些盒子编号。
对于
0
≤
i
≤
m
0\leq i\leq m
0≤i≤m 维护
x
i
‾
x^{\underline i}
xi:
(
x
+
1
)
i
‾
=
x
i
−
1
‾
⋅
(
x
+
1
)
=
(
x
−
i
+
1
)
x
i
−
1
‾
+
i
⋅
x
i
−
1
‾
=
x
i
‾
+
i
⋅
x
i
−
1
‾
\begin{aligned} &(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{aligned}
===(x+1)ixi−1⋅(x+1)(x−i+1)xi−1+i⋅xi−1xi+i⋅xi−1
最后 O ( m ) O(m) O(m) 还原成 x m x^m xm。
总复杂度 O ( m 2 + m n ) O(m^2+mn) O(m2+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;
}