[hdu6201]transaction transaction transaction(树形dp)

题意：某人在一棵树中在某处买物品，价格为i，在某处卖物品，价格为j，每单位距离花费价格1，求最大赚钱数。

解题关键：两次树形dp，分别求出每个点作为被减和被加情况下的最大值，最后取一下max即可。

该节点被减的情况，为他和他所在的子树上的最大值，并且是他的各父节点的被减，该节点被加情况的最大值；

该节点被加的情况，为他和他所在的子树上的最大值，并且是他的各父节点的被加，该节点被减情况的最大值。

 #include<cstdio>
 #include<cstring>
 #include<algorithm>
 #include<cstdlib>
 #include<cmath>
 #include<iostream>
 using namespace std;
 typedef long long ll;
 const int maxn=1e6+;
 const ll inf=1e17+;
 ll head[maxn],tot,n,m,sum,val[maxn];
 struct edge{
     ll to;
     ll nxt;
     ll w;
 }e[maxn<<];
 void add_edge(int u,int v,int w){
     e[tot].to=v;
     e[tot].w=w;
     e[tot].nxt=head[u];
     head[u]=tot++;
 }

 ll dp[maxn],cnt[maxn],ans,dp2[maxn];

 void dfs(ll u,ll fa){
     for(int i=head[u];i!=-;i=e[i].nxt){
         int v=e[i].to;
         int w=e[i].w;
         if(v==fa) continue;
         dfs(v,u);
         dp[u]=max(dp[v]+val[v]-w-val[u],dp[u]);
         dp2[u]=max(dp2[v]-val[v]-w+val[u],dp2[u]);
     }
 }

 inline int read(){
     char k=;char ls;ls=getchar();for(;ls<''||ls>'';k=ls,ls=getchar());
     int x=;for(;ls>=''&&ls<='';ls=getchar())x=(x<<)+(x<<)+ls-'';
     if(k=='-')x=-x;return x;
 }

 int main(){
     int k=;
     int T;
     T=read();
     while(T--){
         n=read();
         memset(dp,,sizeof dp);
         memset(head,-,sizeof head);
         memset(dp2,,sizeof dp2);
         tot=;
         sum=;
         int a,b,c;
         for(int i=;i<=n;i++)  val[i]=read();
         for(int i=;i<n-;i++){
             a=read();
             b=read();
             c=read();
             add_edge(a,b,c);
             add_edge(b,a,c);
         }
         dfs(,-);
         ll ans=-inf;
         for(int i=;i<=n;i++){
             ans=max(ans,dp[i]);
             ans=max(ans,dp2[i]);
         }
         printf("%lld\n",ans);
     }
     return ;
 }