51nod 1424 零树(树dp)

dp【u】【0】代表根节点需要减的数，dp【u】【1】代表根节点需要加的数。

#include <bits\stdc++.h>
using namespace std;
const int maxn=1e5+5;
vector<int>a[maxn];
long long dp[maxn][2];
int v[maxn];
void dfs(int u,int fa)
{
for(int i=0;i<a[u].size();i++)
{
int son=a[u][i];
if(son==fa)continue;
dfs(son,u);
dp[u][0]=max(dp[u][0],dp[son][0]);
dp[u][1]=max(dp[u][1],dp[son][1]);
}
long long tmp=v[u]+dp[u][1]-dp[u][0];
dp[u][tmp<0]+=abs(tmp);
}
int main(){
int n,x,y;
scanf("%d",&n);
memset(dp,0,sizeof(dp));
for(int i=1;i<n;i++)
{scanf("%d%d",&x,&y);
a[x].push_back(y);
a[y].push_back(x);
}
for(int i=1;i<=n;i++)
scanf("%d",&v[i]);
dfs(1,-1);
cout<<(dp[1][0]+dp[1][1])<<endl;
return 0;
}