hdu2196树形dp+邻接表

题目大意：找离每个节点的最远距离

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