D-生活在树上 SDU2020个人排位赛#3

D-生活在树上 SDU2020个人排位赛#3

D. 生活在树上

https://codeforces.com/gym/323026/problem/D

很遗憾，在宿舍使用电饼铛是违规的，zdx并没能成功烙好饼。事实上，他一打开开关，整层楼的电闸都跳了，尽管他已经把线接在了空调专用插口上。zdx需要恢复宿舍的供电，宿舍的结构可以看成一棵有根树，有n个节点，编号从0到n−1，其中0号点是根节点。每个节点有一个数值，zdx每次行动会指定一个点，然后把这个点到根节点的路径上所有节点的数值都+1或者−1。直到所有节点的数值都变为0，宿舍的供电才会恢复。那么，zdx至少要行动多少次

Input
第一行一个整数n,1≤n≤100000，代表节点个数
接下来的第i行(1≤i≤n),每行一个整数，代表第i个节点的父节点
最后一行有n个整数，第i个数|ai|≤100000，代表第i个节点初始数值
保证输入是有效的树

Output
输出一个整数，代表zdx最少的行动次数

Example
inputCopy
3
0
0
2 1 3

outputCopy
6

用带lazy标记的dfs（可能会有点像懒惰标记线段树的思想）。
当前节点数值a加上lazy就是该点子树全部归0后，该点的数值。

#include<bits/stdc++.h>
using namespace std;
vector <int> v[100100];
long long ans=0;
long long a[100100];
long long lazy[100100];
int fa[100100];
void dfs(int n)
{
    if(v[n].size()>0){
        for(auto i:v[n]){
            dfs(i);
        }
    }
    a[n]+=lazy[n];
    ans+=abs(a[n]);
    lazy[fa[n]]+=-a[n]+lazy[n];//这里一定要记得+lazy[n]！！！不然下面的修改传不上去
}
int main()
{
    int tmp,n;
    cin>>n;
    for(int i=1;i<n;i++){
        cin>>tmp;
        v[tmp].push_back(i);
        fa[i]=tmp;
    }
    for(int i=0;i<n;i++){
        cin>>a[i];
    }
    dfs(0);
    cout<<ans;
    return 0;
}