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

时间：2015-05-11 09:02:08 阅读：104 评论：0 收藏：0 [点我收藏+]

标签：

//给一棵树，每次操作可以将包括顶点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;
}

标签：

原文地址：http://blog.csdn.net/cq_pf/article/details/45626245

踩

(0)

评论一句话评论（0）

分享档案

更多>

2021年07月29日 (22)
2021年07月28日 (40)
2021年07月27日 (32)
2021年07月26日 (79)
2021年07月23日 (29)
2021年07月22日 (30)
2021年07月21日 (42)
2021年07月20日 (16)
2021年07月19日 (90)
2021年07月16日 (35)

周排行