题意:
给你一棵树,让你求出以每个节点为根节点 所能到达的最远的距离。
思路:
树形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;
}