2017沈阳网络赛 1008 HDU 6201 transaction transaction transaction（树形dp）

题意：给你一颗树，边有权值，现在有本书，每个点都有这本书的价格，现在你可以选任意一点作为起点买这本书，跑去任意一点卖掉，过程中消耗路径的权值花费，问你任选起点和终点最大收益是多少。

思路：树形DP，dp[u][0]表示以u为根的子树中买一本书的最大收益，dp[u][1]表示以u为根的子树中卖一本书的最大收

益。dp[u][0]+dp[u][1]即为在以u为根的子树中选两点的最大收益。

代码：

#include<bits/stdc++.h>
using namespace std;
typedef long ll;
const int maxn = 1e5+5;
struct node
{
    int v, w;
    node() {}
    node(int vv, int ww): v(vv), w(ww) {}
};
vector<node> g[maxn];
int a[maxn], dp[maxn][2], ans;

void dfs(int u, int pre)
{
    dp[u][0] = -a[u];
    dp[u][1] = a[u];
    for(int i = 0; i < g[u].size(); i++)
    {
        int v = g[u][i].v;
        int w = g[u][i].w;
        if(v == pre) 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);
    }
    ans = max(ans, dp[u][0]+dp[u][1]);
}

int main(void)
{
    int _, n;
    cin >> _;
    while(_--)
    {
        scanf("%d", &n);
        for(int i = 1; i <= n; i++)
            scanf("%d", &a[i]);
        for(int i = 1; i <= n; i++)
            g[i].clear();
        for(int i = 1; i <= n-1; i++)
        {
            int u, v, w;
            scanf("%d%d%d", &u, &v, &w);
            g[u].push_back(node(v, w));
            g[v].push_back(node(u, w));
        }
        ans = 0;
        dfs(1, -1);
        printf("%d\n", ans);
    }
    return 0;
}