P3047 [USACO12FEB]Nearby Cows G

 

一棵 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;
 }