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;
}