题目大意:找离每个节点的最远距离
思路:dp[i][0]表示子树的最远距离,dp[i][1] 表示子树以外的最远距离,dp[i][2] 表示字数第二远距离
第一遍dfs求出dp[i][0]和dp[i][2];第二遍求出dp[i][1],注意第一遍结束后初始化dp[1][1].其他的看代码
#include <iostream>
#include <cstdio>
#include <string>
#include <cstring>
#include <fstream>
#include <algorithm>
#include <cmath>
#include <queue>
#include <stack>
#include <vector>
#include <map>
#include <set>
#include <iomanip>
using namespace std;
//#pragma comment(linker, "/STACK:102400000,102400000")
#define maxn 10005
#define MOD 1000000007
#define mem(a , b) memset(a , b , sizeof(a))
#define LL long long
#define INF 100000000
int n;
struct edge
{
int u , v , w;
int next;
}E[maxn];
int head[maxn];
int dp[maxn][3];
int dfs(int root)
{
if(head[root] < 0) return 0;
for(int i = head[root] ; i >= 0 ; i = E[i].next)
{
int tmp = dfs(E[i].v) + E[i].w;
if(tmp > dp[root][0])
{
dp[root][2] = dp[root][0];
dp[root][0] = tmp ;
}
else if(tmp > dp[root][2]) dp[root][2] = tmp;
}
return dp[root][0];
}
int dfs2(int root)
{
for(int i = head[root] ; i >= 0 ; i = E[i].next)
{
dp[E[i].v][1] = max(dp[root][1] , dp[E[i].v][0] + E[i].w == dp[root][0] ? dp[root][2] : dp[root][0]) + E[i].w;
dfs2(E[i].v);
}
}
int main()
{
while(scanf("%d" , &n)!= EOF)
{
int id = 0;
mem(dp, 0);mem(head , -1);
int u , w;
for(int i = 2 ; i <= n ; i ++)
{
scanf("%d %d" , &u , &w);
int tmp = i ;
if(i > u) swap(u , tmp);
E[id].u = tmp;
E[id].v = u;
E[id].w = w;
E[id].next = head[tmp];
head[tmp] = id++;
}
dfs(1);
dp[1][1] = dp[1][2];
dfs2(1);
for(int i = 1 ; i <= n ; i ++) printf("%d\n" , max(dp[i][0] , dp[i][1]));
}
return 0;
}