比较简单的树形dp(递推？）
设$dp[i][j]$表示距离$i$距离为$j$的点的数目，先预处理$g[i][j]$表示点$i$的子树中距离这个点距离为$j$的点的数目(猫老师讲过，用一个栈维护一下就好了)，然后再预处理根节点，之后开始考虑dp。
当进入一个儿子时，首先这个儿子对于所有距离$dis$会继承距离父亲$dis-1$的点，然后是它的子树中距离它为$dis$的点，但是这样距离它为$dis-2$的点会被算重复，减掉就好了
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int N=,K=;
int p[N],noww[*N],goal[*N];
int num[N],stk[N],pts[N][K],dp[N][K];
int n,k,t1,t2,cnt,top,ans;
void link(int f,int t)
{
noww[++cnt]=p[f];
goal[cnt]=t,p[f]=cnt;
}
void MARK(int nde,int fth)
{
stk[++top]=nde;
for(int i=;i<=k;i++)
if(top>i) pts[stk[top-i]][i]+=num[nde];
for(int i=p[nde];i;i=noww[i])
if(goal[i]!=fth) MARK(goal[i],nde);
top--;
}
void DFS(int nde,int fth)
{
dp[nde][]=pts[nde][];
for(int i=;i<=k;i++)
if(nde==) dp[][i]=pts[][i];
else dp[nde][i]=dp[fth][i-]-pts[nde][i-]+pts[nde][i];
for(int i=p[nde];i;i=noww[i])
if(goal[i]!=fth) DFS(goal[i],nde);
}
int main()
{
scanf("%d%d",&n,&k);
for(int i=;i<n;i++)
{
scanf("%d%d",&t1,&t2);
link(t1,t2),link(t2,t1);
}
for(int i=;i<=n;i++)
scanf("%d",&num[i]);
MARK(,); DFS(,);
for(int i=;i<=n;i++)
{
ans=;
for(int j=;j<=k;j++)
ans+=dp[i][j];
printf("%d\n",ans);
}
return ;
}