HDU 2196 Computer 树形DP

题目大意：

每台新电脑都与某一台原电脑相连有一个长度，求每台电脑相距其最远的电脑的距离

这里因为第一台电脑是最初的，所以可以将第一台电脑作为树根，其他电脑分布就可以形成一棵树

这里距离有两种，一种是往树底找，一种是往父节点方向走

第一次dfs记录下每个节点往子节点方向找到其树底的最长距离，第二次dfs记录每个节点如果一开始往父节点方向走所能得到的最大距离

最后判断两者中得到一个最大值就行了

#include <cstdio>
#include <cstring>
#include <iostream>
using namespace std;

const int N = 10005;

int first[N] , maxn[N][4] , k , vis[N];
/*
maxn[i][0] 表示i往下到其树底的最长距离，
maxn[i][1]表示到树底的次长距离，
maxn[i][2]表示i节点往上寻找到的最长距离
最后只要比较判断往上找还是往下找能得到的最大距离即可
*/
struct Edge{
    int y , next , d;
}e[N<<1];

void add_edge(int x , int y , int d)
{
    e[k].y = y , e[k].d = d , e[k].next = first[x];
    first[x] = k++;
}

void my_swap(int &x , int &y)
{
    int temp = x;
    x = y , y = temp;
}

void dfs1(int u)
{
    vis[u] = 1;
    for(int i = first[u] ; i!=-1 ; i=e[i].next){
        int v = e[i].y;
        if(!vis[v]){
            dfs1(v);
            //更新次长和最长值,自底向上更新，所以写在dfs后面进行回溯
            if(maxn[u][1] < maxn[v][0] + e[i].d){
                maxn[u][1] = maxn[v][0] + e[i].d;
                if(maxn[u][0] < maxn[u][1])
                    my_swap(maxn[u][0] , maxn[u][1]);
            }
        }
    }
}

void dfs2(int u)
{
    vis[u] = 1;
    for(int i= first[u] ; i!=-1 ; i=e[i].next){
        int v = e[i].y;
        if(!vis[v]){
            //自顶向下更新，所以写在dfs前面
            //判断是否最大值落于当前分支
            int tmp;
            if(maxn[u][0] - maxn[v][0] == e[i].d)
                tmp = maxn[u][1];
            else tmp = maxn[u][0];
            maxn[v][2] = max(maxn[u][2] , tmp) + e[i].d;
            dfs2(v);
        }
    }
}

int main()
{
   // freopen("a.in" , "r" , stdin);
    int n , d , a;
    while(scanf("%d" , &n) == 1)
    {
        k = 0;
        memset(first , -1 , sizeof(first));
        for(int i=2 ; i<=n ; i++){
            scanf("%d%d" , &a , &d);
            add_edge(a , i , d);
            add_edge(i , a , d);
        }

        memset(maxn , 0 , sizeof(maxn));
        memset(vis , 0 , sizeof(vis));
        dfs1(1);

        memset(vis , 0 , sizeof(vis));
        dfs2(1);
        /*
        for(int i=1 ; i<=n ; i++)
            printf("%d %d %d
" , maxn[i][0] , maxn[i][1] , maxn[i][2]);
        */
        for(int i=1 ; i<=n ; i++)
            printf("%d
" , max(maxn[i][0] , maxn[i][2]));
    }
    return 0;
}