水题
19：46开始看题，20：09AC
就是大力dp啊，设dp[i][j]表示考虑到第i个点，距离它为j的数量，第一次倒推，求出子树中的，第二次正推，从父节点转移
转移见代码
// luogu-judger-enable-o2
#include<iostream>
#include<cstdio>
#include<queue>
#include<algorithm>
#include<cstring>
#include<cmath>
#include<set>
#include<map>
#include<vector>
#define maxn 400006
#define int long long
#define SZJ signed
#include<time.h>
#define AK main
#define half (l+r)>>1
#define SDOI ()
#define mod 19650827
using namespace std;
#define rep(i,a,b) for (int (i)=(a);i<=(b);++(i))
#define dep(i,a,b) for (int (i)=(a);i>=(b);--(i))
#define erep(i,a) for (int (i)=head[(a)];(i)!=-1;(i)=e[(i)].next)
#define lson t[s].lc
#define rson t[s].rc
#define MP make_pair
int head[maxn],cur,n,k,dp[maxn][25],size[maxn],val[maxn];
struct hzw
{
int to,next;
}e[maxn];
inline void add(int a,int b)
{
e[cur].to=b;
e[cur].next=head[a];
head[a]=cur++;
}
inline void dfs(int s,int fa)
{
size[s]=1;
dp[s][0]=val[s];
erep(i,s)
{
if (e[i].to==fa) continue;
dfs(e[i].to,s);
size[s]+=size[e[i].to];
rep(j,1,k) dp[s][j]+=dp[e[i].to][j-1];
}
}
int bef[25];
inline void dfs2(int s,int fa)
{
if (s!=1)
{
rep(i,0,k) bef[i]=dp[s][i];
dp[s][1]+=val[fa];
rep(i,2,k)
{
dp[s][i]+=dp[fa][i-1];
dp[s][i]-=bef[i-2];
}
}
erep(i,s)
{
if (e[i].to==fa) continue;
dfs2(e[i].to,s);
}
}
int fina[maxn];
SZJ AK SDOI
{
memset(head,-1,sizeof(head));
cin>>n>>k;
rep(i,1,n-1)
{
int a,b;
scanf("%lld%lld",&a,&b);
add(a,b);
add(b,a);
}
rep(i,1,n) scanf("%lld",&val[i]);
dfs(1,1);
dfs2(1,1);
rep(i,1,n) rep(j,0,k) fina[i]+=dp[i][j];
rep(i,1,n) printf("%lld
",fina[i]);
}