马风很好,代码如下:
#include<bits/stdc++.h>
using namespace std;
int n,k,dp[100001][21],ans[100001][21],c[100001],fa[100001];
vector<int> w[100001],v[100001];
bool vis[100001];
void build(int now){
vis[now]=1;
for(int i=0;i<w[now].size();i++){
int t=w[now][i];
if(!vis[t]){
vis[t]=1;
v[now].push_back(t);
fa[t]=now;
build(t);
}
}
}
void dfs(int now){
for(int i=0;i<v[now].size();i++) dfs(v[now][i]);
dp[now][0]=c[now];
for(int i=0;i<v[now].size();i++){
int t=v[now][i];
for(int j=1;j<=k;j++) dp[now][j]+=dp[t][j-1];
}
}
void dfs2(int now){
ans[now][0]=c[now];
for(int i=1;i<=k;i++){
if(fa[now]){
if(i==1) ans[now][i]=dp[now][i]+c[fa[i]];
else ans[now][i]=ans[fa[now]][i-1]-dp[now][i-2]+dp[now][i];
}else ans[now][i]=dp[now][i];
}
for(int i=0;i<v[now].size();i++) dfs2(v[now][i]);
}
int main(){
scanf("%d%d",&n,&k);
for(int i=1;i<n;i++){
int u,v;
scanf("%d%d",&u,&v);
w[u].push_back(v);
w[v].push_back(u);
}
for(int i=1;i<=n;i++) scanf("%d",&c[i]);
build(1);
dfs(1);
for(int i=1;i<=n;i++){
for(int j=1;j<=k;j++) dp[i][j]+=dp[i][j-1];
}
dfs2(1);
for(int i=1;i<=n;i++) printf("%d\n",ans[i][k]);
return 0;
}