51 nod 1424

1424 零树

有一棵以1为根的树，他有n个结点，用1到n编号。第i号点有一个值vi。

现在可以对树进行如下操作：

步骤1：在树中选一个连通块，这个连通块必须包含1这个结点。

步骤2：然后对这个连通块中所有结点的值加1或者减1。

问最少要经过几次操作才能把树中所有结点都变成0。

注意：步骤1与步骤2合在一起为一次操作。

 

Input

单组测试数据。

第一行有一个整数n(1 ≤ n ≤ 10^5)

接下来n-1行，每行给出 ai 和 bi (1 ≤ ai, bi ≤ n; ai ≠ bi)，表示ai和bi之间有一条边，输入保证是一棵树。

最后一行有n个以空格分开的整数，表示n个结点的值v1, v2, ..., vn (|vi| ≤ 10^9)。

Output

输出一个整数表示最少的操作步数。.

Input示例

3

1 2

1 3

1 -1 1

Output示例

3

System Message (题目提供者)

C++的运行时限为：1000 ms ，空间限制为：131072 KB  

题解：对于树来说，任意一个节点如果与根节点联通，那么其祖先一定和根节点联通。所以对于任意一颗子树，up【i】表示以i为根的子树最少+1的次数，down【i】表示i为根的子树最少-1的次数，每次取儿子节点中最大的即可。

#include <iostream>
#include <cstring>
#include <cstdio>
#include <algorithm>
#define N 300000
using namespace std;
int n;
int last[N<<1],to[N<<1],head[N],cnt=0;
void ins(int u,int v){
	last[++cnt]=head[u];head[u]=cnt;to[cnt]=v;
}
long long down[N],up[N],val[N]; 
void dfs(int x,int fa)
{
	for(int i=head[x];i;i=last[i])
	{
		if(to[i]==fa) continue;
		dfs(to[i],x);
		down[x]=max(down[x],down[to[i]]);
		up[x]=max(up[x],up[to[i]]);
	}
	val[x]+=(up[x]-down[x]);
	if(val[x]>0) down[x]+=val[x];
	if(val[x]<0) up[x]-=val[x];
	return ;
}
int main()
{
	int u,v;
	scanf("%d",&n);
	for(int i=1;i<n;i++)
	{
		scanf("%d%d",&u,&v);
		ins(u,v);ins(v,u);
	}
	for(int i=1;i<=n;i++) scanf("%lld",&val[i]);
	dfs(1,0);
	printf("%lld",(up[1]+down[1]));
	return 0;
}