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是这样的）

 

 

?

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49

#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;  }