男神的树

男神的树

题目链接：http://acm.xidian.edu.cn/problem.php?id=1164

DFS

早上才写了一题...回来发现除了这题外其他都好简单= =

这题用vector存孩子会爆空间，所以想到用father[N]存父节点，lazy[N]数组存修改值（和线段树的lazy用法一样），vis[N]标记是否被dfs，再用dfs向上搞

//其实数据比较弱1e6就可以过，lazy没用LL好像也没问题，后来改了发1e6的，没改名次(╯‵□′)╯︵┻━┻耗时更长了什么鬼...

//这样做内存消耗125M，差不多要挂了= =

代码如下：

#include<cstdio>
#include<cmath>
#include<vector>
#define N (int)(1e7+5)
using namespace std;
typedef long long LL;
int father[N],value[N],lazy[N];
bool vis[N];
int n,u,v;
LL sum;
LL dfs(int index){
    if(vis[index])return 0;
    vis[index]=1;
    if(father[index]==0){
        lazy[index]-=value[index];
        return abs(lazy[index]);
    }
    LL s=dfs(father[index]);
    value[index]+=lazy[father[index]];
    lazy[index]+=(lazy[father[index]]-value[index]);
    s+=abs(value[index]);
    return s;
}
int main(void){
    scanf("%d",&n);
    for(int i=0;i<n-1;++i){
        scanf("%d%d",&u,&v);
        father[v]=u;
    }
    for(int i=1;i<=n;++i)
        scanf("%d",&value[i]);
    for(int i=1;i<=n;++i)
        sum+=dfs(i);
    printf("%lld\n",sum);
}