题意:给你一颗树,边有权值,现在有本书,每个点都有这本书的价格,现在你可以选任意一点作为起点买这本书,跑去任意一点卖掉,过程中消耗路径的权值花费,问你任选起点和终点最大收益是多少。
思路:树形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;
}