51nod - 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

思路：树形DP

因为每次都需要经过1节点，所以考虑要先满足叶子节点能减为0，然后再逐级往上。因此需要用到DFS。

#include <iostream>    
#include <cstring>    
#include <vector>    
using namespace std; 
 
const int MAXN = 1e5 + 5;
typedef long long ll;    
  
vector<int> line[MAXN];  
ll num[MAXN],  up[MAXN], down[MAXN]; 

void dfs(int cur, int father)
{
	for (int i = 0; i < line[cur].size(); i++)
	{
		int peer = line[cur][i];
		if (peer != father)
		{
			dfs(peer, cur);
			up[cur] = max(up[cur], up[peer]);
			down[cur] = max(down[cur], down[peer]);
		}
	}

	num[cur] += up[cur] - down[cur];
	if (num[cur] > 0)
	{
		down[cur] += num[cur];
	}
	else if (num[cur] < 0)
	{
		up[cur] -= num[cur];
	}
}

int main()
{  
    int n, a, b;  
    cin >> n;
    for (int i = 0; i < n - 1; i++)
    {  
        cin >> a >> b;
        line[a].push_back(b);  
        line[b].push_back(a);  
    }

    for (int i = 1; i <= n; i++)  
    {
        cin >> num[i];
    }
    dfs(1, -1);  
    cout << up[1] + down[1] << endl;
    
    return 0;  
}