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;}