一棵 n个点的树,点带权,对于每个节点求出距离它不超过 m 的所有节点权值和 S[i ]
换根dp
1. f[i][L] += f[y][L-1]
2 . g[i][L] =f[i][L] ,
g[i][L]+= g[fa][L-1] -f[y][L-2] (这个脑补一下图片, 换根dp是这样的)
#include <bits/stdc++.h> using namespace std ; const int N=1e5+3,M=N*4; int nxt[M],go[M],hd[N],all; int n,m,a[N],f[N][22],g[N][22]; void add(int x,int y){ all++; go[all]=y,nxt[all]=hd[x],hd[x]=all; } void dfs1(int x,int father){ int i; for(i=0;i<=m;i++) f[x][i]=a[x]; for(i=hd[x];i;i=nxt[i]){ int y=go[i]; if(y==father) continue; dfs1(y,x); for(int j=1;j<=m;j++) f[x][j]+=f[y][j-1]; } } void dfs(int x,int father){ for(int i=hd[x];i;i=nxt[i]){ int y=go[i]; if(y==father) continue; g[y][1]+=f[x][0]; for(int j=2;j<=m;j++) g[y][j]+=g[x][j-1]-f[y][j-2]; dfs(y,x); } } signed main(){ cin>>n>>m; int x,y; for(int i=1;i<n;i++) cin>>x>>y,add(x,y),add(y,x); for(int i=1;i<=n;i++) cin>>a[i]; dfs1(1,0); for(x=1;x<=n;x++) for(int i=0;i<=m;i++) g[x][i]=f[x][i]; dfs(1,0); for(int i=1;i<=n;i++) cout<<g[i][m]<<endl; }