90pts Wa on#5玄关求助

#include<bits/stdc++.h>
#define int long long
#define mod (int)(1e9+7)
#define N (int)(1e5+10)
#define K 25
using namespace std;
int n,k;
int f[N][K];
int g[N][K];
int a[N];
vector<int>e[N];
int vis[N];
void dfs(int x,int fa){
	if(vis[x])return;
	vis[x]=1;
	f[x][0]=a[x];
	if(e[x].size()==0)return;
	for(int j=1;j<=k;j++){
		f[x][j]=a[x];
		for(int i=0;i<e[x].size();i++){
			int y=e[x][i];
			if(y==fa)continue;
			dfs(y,x);
			f[x][j]+=f[y][j-1];
		}
	}
}
void _dfs(int x,int fa){
	if(x!=1){
		g[x][0]=a[fa];
		for(int i=1;i<k;i++){
			g[x][i]=g[fa][i-1]+f[fa][i]-f[x][i-1];
		}
	}
	for(int i=0;i<e[x].size();i++)if(e[x][i]!=fa)_dfs(e[x][i],x);
}
signed main(){
	cin>>n>>k;
	for(int i=1;i<n;i++){
		int u,v;
		cin>>u>>v;
		e[u].push_back(v);
		e[v].push_back(u);
	}
	for(int i=1;i<=n;i++)cin>>a[i];
	dfs(1,0);
	_dfs(1,0);
	for(int i=1;i<=n;i++){
		cout<<f[i][k]+g[i][k-1];
		cout<<"\n";
	}
    return 0;
}