codeforces 168 div2 Zero Tree

http://codeforces.com/contest/275/problem/D

发点题证明自己在做题

对于每个节点 记录他的孩子节点中最大的减少的值 和 最大的增加值 执行到该点的时候必须执行的步骤是key-sub+add

代码：

#include <stdio.h>
#include <stdlib.h>
#include <math.h>
#include <string.h>
#include <iostream>
#include <algorithm>
#include <map>
#include <set>
#include <queue>
#include <vector>
#include <iomanip>
using namespace std;
vector<int> h[100005];
struct node
{
    long long key;
    long long sub;
    long long add;
}
num[100005];
long long ans;
long long node;
long long absl(long long a)
{
    if(a<0)
    return -a;
    return a;
}
void dfs(int x,int fa)
{
    int len=h[x].size();
    int L=len;
    if(x==1)
    len++;
    if(len==1)
    return;
    for(int i=0;i<L;i++)
    {
        if(h[x][i]==fa)
        continue;
        dfs(h[x][i],x);
        int son=h[x][i];
        num[x].sub=max(num[son].sub,num[x].sub);
        num[x].add=max(num[son].add,num[x].add);
        if(num[son].key>0)
        num[x].sub=max(num[son].sub+num[son].key,num[x].sub);
        else
        num[x].add=max(num[son].add-num[son].key,num[x].add);
    }
    num[x].key=num[x].key-num[x].sub+num[x].add;
    //printf("%I64d %I64d %I64d\n",num[x].key,num[x].sub,num[x].add);
}
int main()
{
    int n;
    scanf("%d",&n);
    memset(num,0,sizeof(num));
    for(int i=1;i<=n;i++)
    h[i].clear();
    for(int i=1;i<n;i++)
    {
        int fa,son;
        scanf("%d%d",&fa,&son);
        h[fa].push_back(son);
        h[son].push_back(fa);
    }
    for(int i=1;i<=n;i++)
    {
        long long key;
        scanf("%I64d",&key);
        num[i].key=key;
    }
    dfs(1,-1);
    printf("%I64d\n",num[1].sub+num[1].add+absl(num[1].key));
}