【zcmu1920】nazime's 友人帐(贪心,树形dp)

记录一个菜逼的成长。。

题目链接
原题

题目没说根是多少。
但是连通块必须包含1，我们不妨以1为根。

1.假设有个点i，那么如果以点i为根的子树上有点包含在所选取的块，那么也必定会对i这个点进行同样的操作。因为到1的路径中必定包含i。
2.所以点i的操作次数就是点i的儿子节点中的最大操作次数
(因为如果有两个或以上的结点需要相同的操作，可以合并操作)
(对于叶子结点，如果权值为正down的值就是权值，否则up的值就是权值的绝对值)

对于up来说，
即

$$up[i] = max \space up[j] (j\space adjacent\space to\space i)$$
$down[i]$同理
因为每次修改1结点必定会被修改，所以 $ans = down[1] + up[1]$
那么只要dfs一遍，回溯更新即可
时间复杂度 $O(n)$

#include <bits/stdc++.h>
using namespace std;
#define pb push_back
#define mp make_pair
typedef long long LL;
const int maxn = 100000 + 10;
vector<int>edge[maxn];
LL a[maxn],down[maxn],up[maxn];
void dfs(int u,int pre)
{
    up[u] = down[u] = 0;
    for( int i = 0; i < edge[u].size(); i++ ){
        int v = edge[u][i];
        if(v == pre)continue;
        dfs(v,u);
        down[u] = max(down[u],down[v]);
        up[u] = max(up[u],up[v]);
    }
    a[u] += up[u] - down[u];
    if(a[u] < 0)up[u] -= a[u];
    else down[u] += a[u];
}
int main()
{
    int n;
    while(~scanf("%d",&n)){
        for( int i = 0; i < n-1; i++ ){
            int u,v;
            scanf("%d%d",&u,&v);
            edge[u].pb(v);
            edge[v].pb(u);
        }
        for( int i = 1; i <= n; i++ )scanf("%lld",a+i);
        dfs(1,-1);
        printf("%lld\n",up[1] + down[1]);
    }
    return 0;
}