51nod1424 零数

最新推荐文章于 2022-02-26 16:52:46 发布

LL_Sagiri

最新推荐文章于 2022-02-26 16:52:46 发布

阅读量158

点赞数

CC 4.0 BY-SA版权

分类专栏： 51nod 其它

本文链接：https://blog.youkuaiyun.com/qq_41510496/article/details/79491202

51nod 同时被 2 个专栏收录

51 篇文章

订阅专栏

其它

4 篇文章

订阅专栏

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示例

Output示例

System Message (题目提供者)

题解：精妙的想法。

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