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

最新推荐文章于 2017-10-02 21:27:34 发布

原创最新推荐文章于 2017-10-02 21:27:34 发布 · 657 阅读

1 ·

本内容遵循CC 4.0 BY-SA版权协议

标签

#沈阳网络赛 #HDU #dp

DP 专栏收录该内容

94 篇文章

订阅专栏

本文介绍了一种使用树形动态规划解决特定问题的方法：在一棵树上，通过选择最佳的购书和售书节点来最大化收益。代码示例展示了如何初始化数据结构、递归遍历树并更新状态以找到全局最优解。

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

思路：树形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;
}