Codeforces Round #168 (Div. 1)B 树形dp

//给一棵树，每次操作可以将包括顶点1的连通子集的所有点的节点加1或减1

//问最少几次操作使得这棵树的所有顶点的值都为0

//以1为根节点建树

//将加和减分开考虑,用up[u],down[u]表示以u为跟节点的子树中需要加的操作

//最大为up[u] ,需要减的操作最大为down[u]

//其余的加和减的操作则可以在处理这两个操作时一起覆盖

//在将u的子数全都处理完后u点的值由于在加了up[u]和减了down[u]后变为

//一个新的值，则对于这个新的值根据它的正负并入up[u]或down[u]

//直到更新到1节点结束

#include<cstdio>

#include<cstring>

#include<iostream>

using namespace std ;

const int maxn = 100010 ;

typedef __int64 ll;

ll dp[maxn] ;

int vis[maxn] ;

ll up[maxn] , down[maxn] ;

struct Edge

{

int  v;

int next ;

}edge[maxn*2] ;

int head[maxn] ;

int nedge ;

void addedge(int u , int v)

{

edge[nedge].v = v;

edge[nedge].next = head[u] ;

head[u] = nedge++;

edge[nedge].v = u ;

edge[nedge].next = head[v] ;

head[v] = nedge++ ;

}

void dfs(int u)

{

vis[u] = 1;

for(int i = head[u];i != -1 ;i = edge[i].next)

{

int v = edge[i].v;

if(vis[v]) continue ;

dfs(v) ;

up[u] = max(up[v] , up[u]) ;

down[u] = max(down[u] , down[v]) ;

}

ll t = dp[u] + up[u] - down[u] ;

if(t > 0) down[u] += t ;

else up[u] -= t;

}

void init()

{

memset(up , 0 , sizeof(up)) ;

memset(down , 0 , sizeof(down));

memset(vis , 0 ,sizeof(vis)) ;

memset(head , -1 , sizeof(head)) ;

nedge = 0 ;

}

int main()

{

//freopen("input.txt" ,"r" ,stdin) ;

int n ;

while(~scanf("%d" ,&n))

{

int u , v;

init() ;

for(int i = 1;i < n;i++)

{

scanf("%d%d" ,&u , &v) ;

addedge(u , v) ;

}

for(int i = 1;i <= n;i++)

scanf("%I64d" ,&dp[i]) ;

dfs(1) ;

printf("%I64d\n" ,up[1] + down[1]) ;

}

return 0;

}