Description
給出一棵n個點的樹,每個點上有C_i頭牛,問每個點k步範圍內各有多少頭牛。
這道題分爲兩部分
第一步
先轉化成有根樹,然後再計算出以每個點爲根的子樹中 往下 走j步得到的牛的總數。
樹形DP完成,弱智公式
f[x][j]=∑u∈sonxf[u][j−1]+f[x][0]
時間複雜度O(n×k) 第二步
每個點往下的已經算好了,接下來就算往上以及拐來拐去的。反正k 不大,暴力往上跳k 步就行了,跳的時候統計。
如何統計呢?
假設現在跳了j 步到達點u ,那麼再加上f[u][k−j] ,有一部分被重複加了,再減掉,就是f[sonu][k−j−1] ,其中sonu 代表在往上跳的過程中經過的u 的兒子,不難發現是唯一的。
時間複雜度O(n×k)
#include <bits/stdc++.h>
using namespace std ;
void Read ( int &x, char c = getchar() ) {
for ( x = 0 ; !isdigit(c) ; c = getchar() ) ;
for ( ; isdigit(c) ; c = getchar() ) x = 10*x + c - '0' ;
}
const int maxn = 1e5+5, maxm = 25 ;
int n, m, f[maxn][maxm] ;
int e, be[maxn], nxt[maxn<<1], to[maxn<<1], fa[maxn] ;
void add ( int x, int y ) {
to[++e] = y ;
nxt[e] = be[x] ;
be[x] = e ;
}
void dfs ( int x ) {
int i, u, j ;
for ( i = be[x] ; i ; i = nxt[i] ) {
u = to[i] ;
if ( u == fa[x] ) continue ;
fa[u] = x ;
dfs(u) ;
}
for ( j = 1 ; j <= m ; j ++ ) {
f[x][j] = f[x][0] ;
for ( i = be[x] ; i ; i = nxt[i] )
if ( to[i] != fa[x] ) f[x][j] += f[to[i]][j-1] ;
}
}
int calc ( int x ) {
int i, u, v = 0, stp = m ;
v = f[x][m] ;
for ( u = fa[x] ; fa[x] && stp>=0 ; x = u, u = fa[x], stp-- ) {
v += f[u][stp-1]-f[x][stp-2] ;
}
return v ;
}
int main() {
int i, j, k, x, y ;
Read(n) ; Read(m) ;
for ( i = 1 ; i < n ; i ++ ) {
Read(x) ; Read(y) ;
add ( x, y ) ;
add ( y, x ) ;
}
for ( i = 1 ; i <= n ; i ++ ) Read(f[i][0]) ;
dfs(1) ;
for ( i = 1 ; i <= n ; i ++ ) printf ( "%d\n", calc(i) ) ;
return 0 ;
}