只AC #1 #5
我这份代码到底哪里错了...看了114514年了

主要问题在dfs2打斜杠那里

我这样写和循环顺序也没关系啊

//f[i][j]表示以i为根的子树走了j步及以内的权值和，
//ans[i][j]表示以i为根的整棵树走了j步及以内的权值和，即答案
//g是存图的
#include<bits/stdc++.h>
#define pb push_back
#define int long long
using namespace std;
const int N=1e5+5;
int n,m,x,y,a[N],f[N][21],son[N],ans[N][21];
vector<int> g[N];
void dfs1(int x,int y){
	for(int i=0;i<g[x].size();i++){
		int j=g[x][i];
		ans[x][1]+=a[j];
		if(j==y) continue;
		dfs1(j,x);
		for(int k=1;k<=m;k++){
			f[x][k]+=f[j][k-1];
		}
	}
}
void dfs2(int x,int y){
	for(int i=0;i<g[x].size();i++){
		int j=g[x][i];
		if(j==y) continue;
		for(int k=2;k<=m;k++)// 
			ans[j][k]=f[j][k]+ans[x][k-1]-f[j][k-2];// 
		dfs2(j,x);
	}
}
signed main(){
	scanf("%lld%lld",&n,&m);
	for(int i=1;i<n;i++){
		scanf("%lld%lld",&x,&y);
		g[x].pb(y); g[y].pb(x);
	}
	for(int i=1;i<=n;i++){
		scanf("%lld",&a[i]);
		for(int j=0;j<=m;j++) f[i][j]=a[i];
		ans[i][0]=ans[i][1]=a[i];
	}
	dfs1(1,0);
	dfs2(1,0);
	for(int i=1;i<=n;i++)
		printf("%lld\n",(i==1?f[i][m]:ans[i][m]));
	return 0;
}