51nod-1424 零树

原题链接

1424 零树

题目来源：  CodeForces

基准时间限制：1 秒 空间限制：131072 KB 分值: 80  难度：5级算法题

 收藏

 关注

有一棵以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

用up[j]表示j节点加1操作的数目，down[j]表示j节点减1操作的数目，up[j] = max(up[h])(h时j的孩子), down[j] = max(down[h])(h时j的孩子)

#include <iostream>  
#include <cstdio>  
#include <cstring>  
#include <algorithm>  
#include <vector>  
#include <queue>  
#include <cmath>
#define maxn 100005  
#define MOD 1000000007  
using namespace std;  
typedef long long ll;  

vector<int> v[maxn];
ll num[maxn],  up[maxn], down[maxn];
void dfs(int j, int f){
	for(int i = 0; i < v[j].size(); i++){
		int h = v[j][i];
		if(h != f){
			dfs(h, j);
			up[j] = max(up[j], up[h]);
			down[j] = max(down[j], down[h]);
		}
	}
	num[j] += up[j] - down[j];
	if(num[j] > 0)down[j] += num[j];
	if(num[j] < 0)up[j] -= num[j];
}
int main(){
	int n, a, b;
	scanf("%d", &n);
	for(int i = 0; i < n - 1; i++){
		scanf("%d%d", &a, &b);
		v[a].push_back(b);
		v[b].push_back(a);
	}
	for(int i = 1; i <= n; i++)
	  scanf("%I64d", num+i);
	dfs(1, -1);
	printf("%I64d\n", up[1] + down[1]);
	return 0;
}