HDU 2196 Computer | 树形dp

最新推荐文章于 2020-12-17 15:19:32 发布

AC_vitasoy

最新推荐文章于 2020-12-17 15:19:32 发布

阅读量406

点赞数

分类专栏：动态规划——普通dp 动态规划——树形dp 文章标签： HDU 2196 Computer

本文链接：https://blog.csdn.net/u011580493/article/details/39156951

版权

动态规划——普通dp 同时被 2 个专栏收录

33 篇文章 0 订阅

订阅专栏

动态规划——树形dp

9 篇文章 0 订阅

订阅专栏

题意：

给你一棵树，让你求出以每个节点为根节点所能到达的最远的距离。

思路：

树形dp问题。

先dfs求出每个点到该点子树的最大距离和第二大距离（此时保存该点的最大距离，次大距离来源的直接儿子节点，下次更新会用到。）

再dfs更新每个点所到达整棵树的最大距离。

AC代码：

#include <cstring>
#include <cstdlib>
#include <iostream>
#include <cstdio>
#include <vector>
using namespace std;
typedef long long ll;
const int MAXN = 1e4+5;
const int INF = 0x3f3f3f3f;
struct Edge
{
        int v, len;
};
vector <Edge> gra[MAXN];
int n;
ll dp[MAXN][2], dis[MAXN][2];
int from[MAXN][2], res[MAXN];

void init()
{
        memset(dis, 0, sizeof(dis));
        memset(from, 0 ,sizeof(from));
        for(int i = 1;i <= n; i++)
                gra[i].clear();
}
void dfs(int u)
{
        dp[u][1] = dp[u][0] = 0;
        for(int i = 0;i < gra[u].size(); i++)
        {
                Edge &e = gra[u][i];
                dfs(e.v);
                if(dp[u][0] < dp[e.v][1] + e.len)
                {
                        dp[u][0] = dp[e.v][1]+e.len;
                        from[u][0] = e.v;
                        if(dp[u][1] < dp[u][0])
                        {
                                swap(dp[u][1], dp[u][0]);
                                swap(from[u][1], from[u][0]);
                        }
                }
        }
}
void solve(int u, ll val)
{
        res[u] = max(dp[u][1], val);
        for(int i = 0; i < gra[u].size(); i++)
        {
                Edge &e = gra[u][i];
                if(e.v == from[u][1])
                        solve(e.v, max(dp[u][0], val)+e.len);
                else
                        solve(e.v, max(dp[u][1], val)+e.len);
        }
}

int main()
{
        while(scanf("%d", &n) != EOF)
        {
                init();
                int u, len;
                for(int i = 2;i <= n; i++)
                {
                        scanf("%d%d", &u, &len);
                        gra[u].push_back((Edge){i, len});
                }
                //
                dfs(1);
                solve(1, 0);
                for(int i = 1;i <= n; i++)
                        printf("%I64d\n", res[i]);
        }
        return 0;
}