hdu 2196 computer （树状DP+dfs）

http://acm.hdu.edu.cn/showproblem.php?pid=2196
题意：有编号1-n的电脑相连成树状，1为根，求每个节点到其它节点的最大距离（n<=10000）
想了好久没想出来怎么DP，后来还是参考的别人的思路和代码。
可以先对每个节点dfs出到叶子节点的最大距离和次大距离，在从根节点向下dfs该节点经过某点的最大距离（可能表达不清楚，看代码吧。。。）
出处：http://www.cnblogs.com/kuangbin/archive/2012/08/28/2659915.html
附代码：
#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>
using namespace std;
#define MAXN 10100
struct Edge
{
    int len;
    int next;
    int to;
}edge[MAXN*2];//这里边要开两倍 
int head[MAXN];
int fmax[MAXN];//最大距离
int smax[MAXN];//次大距离
int ans[MAXN];
int fmaxid[MAXN];
int smaxid[MAXN];
int tot = 0;
void addedge(int a,int b,int dis)
{
    edge[tot].to = b;
    edge[tot].len = dis;
    edge[tot].next = head[a];
    head[a] = tot++;
    edge[tot].to = a;
    edge[tot].len = dis;
    edge[tot].next = head[b];
    head[b] = tot++;


}


void dfs1(int v,int p)
{
    for(int i=head[v];i!=-1;i=edge[i].next)
    {
        int u = edge[i].to;
        if(u == p)continue;//这个不能掉 
        dfs1(u,v);
        if(smax[v]<fmax[u]+edge[i].len)
        {
            smax[v] = fmax[u]+edge[i].len;
            smaxid[v] = u;
            if(smax[v]>fmax[v])
            {
                swap(smax[v],fmax[v]);
                swap(smaxid[v],fmaxid[v]);
            }
        }




    }
}


void dfs2(int v,int p)
{
    for(int i=head[v];i!=-1;i=edge[i].next)
    {
        int u = edge[i].to;
        if(u == p)continue;
        if(fmaxid[v] == u)
        {
            if(smax[u]<smax[v]+edge[i].len)
            {
                smax[u] = smax[v]+edge[i].len;
                smaxid[u] = v;
                if(smax[u]>fmax[u])
                {
                    swap(smax[u],fmax[u]);
                    swap(smaxid[u],fmaxid[u]);
                }
            }
        }
        else
        {
            if(smax[u]<fmax[v]+edge[i].len)
            {
                smax[u] = fmax[v]+edge[i].len;
                smaxid[u] = v;//这里写搓了。。WA了N久 
                if(smax[u]>fmax[u])
                {
                    swap(smax[u],fmax[u]);
                    swap(smaxid[u],fmaxid[u]);
                }
            }
        }
        dfs2(u,v);
    }
}


int main()
{
    int n;
    while(scanf("%d",&n)!=EOF)
    {
        memset(fmax,0,sizeof(fmax));
        memset(smax,0,sizeof(smax));
        memset(head,-1,sizeof(head));
        tot = 0;
        for(int i=2;i<=n;i++)
        {
            int e,len;
            scanf("%d%d",&e,&len);
            addedge(i,e,len);
        }
        dfs1(1,-1);
        dfs2(1,-1);
        for(int i=1;i<=n;i++)
                printf("%d\n",fmax[i]);


    }
    return 0;
}