HDU-6201 transaction transaction transaction（树dp / 最长（短）路）

题目：http://acm.hdu.edu.cn/showproblem.php?pid=6201
题意：n个城市，n-1条路，商人要选一个城市买书，一个城市卖书，每个城市都有对应的price，每条路都有对应的的花费，求最大的money
思路：
树dp，dp[u][0]代表买书最大收益，dp[u][1]代表卖书最大收益
dp[u][0] = max(dp[u][0],dp[v][0]-w);
dp[u][1] = max(dp[u][1],dp[v][1]-w);
最长路（最短路）参考http://www.cnblogs.com/Blackops/p/7501966.html
代码：

#include<bits/stdc++.h>using namespace std;typedef long long ll;const int N = 1e5+5;struct edge{    int v,w;};vector<edge> G[N];int maxn,dp[N][2],p[N];void dfs(int u,int fa){    dp[u][0] = -p[u];    dp[u][1] = p[u];    for(auto e : G[u])    {        int v = e.v,w = e.w;        if(v == fa)            continue;        dfs(v,u);        dp[u][0] = max(dp[u][0],dp[v][0]-w);        dp[u][1] = max(dp[u][1],dp[v][1]-w);    }    maxn = max(maxn,dp[u][0]+dp[u][1]);}int main(){    int T,n;    scanf("%d",&T);    while(T--)    {        scanf("%d",&n);        for(int i = 1;i <= n;i++)            G[i].clear();        for(int i = 1;i <= n;i++)            scanf("%d",&p[i]);        --n;        while(n--)        {            int u,v,w;            scanf("%d%d%d",&u,&v,&w);            G[u].push_back({v,w});            G[v].push_back({u,w});        }        maxn = -0x3f3f3f3f;        dfs(1,-1);        printf("%d\n",maxn);    }    return 0;}