hdu2196 树形dp

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
using namespace std;
const int maxn=10005;

struct Node
{
    int v;
    int len;
    int next;
} node[maxn<<1];

int head[maxn];
int fm[maxn],fn[maxn]; ///最远距离，以及对应的子树的序号
int sm[maxn],sn[maxn];  ///次远距离，以及其对应的子树
int k;

void addedge(int u,int v,int l)
{
    node[k].v=v;
    node[k].len=l;
    node[k].next=head[u];
    head[u]=k++;

    node[k].v=u;
    node[k].len=l;
    node[k].next=head[v];
    head[v]=k++;
}

void dfs1(int u,int p)      ///第一次dfs找到子树种最大的长度，和最大长度的子树的序号
{
    fm[u]=0;   ///初始化
    sm[u]=0;
    for(int i=head[u]; i!=-1; i=node[i].next)
    {
        int v=node[i].v;
        if(v==p) continue;
        dfs1(v,u);
        if(node[i].len+fm[v]>sm[u])
        {
            sm[u]=node[i].len+fm[v];
            sn[u]=v;
            if(sm[u]>fm[u])
            {
                swap(fm[u],sm[u]);
                swap(fn[u],sn[u]);
            }
        }
    }
}

void dfs2(int u,int p)
{
    for(int i=head[u]; i!=-1; i=node[i].next)
    {
        int v=node[i].v;
        if(v==p) continue;
        if(v==fn[u])
        {
            if(node[i].len+sm[u]>sm[v])
            {
                sm[v]=node[i].len+sm[u];
                sn[v]=u;
                if(sm[v]>fm[v])
                {
                    swap(fm[v],sm[v]);
                    swap(fn[v],sn[v]);
                }
            }
        }
        else
        {
            if(node[i].len+fm[u]>sm[v])
            {
                sm[v]=node[i].len+fm[u];
                sn[v]=u;
                if(sm[v]>fm[v])
                {
                    swap(fm[v],sm[v]);
                    swap(fn[v],sn[v]);
                }
            }
        }
        dfs2(v,u);
    }
}

int main()
{
    int n;
    while(scanf("%d",&n)==1)
    {
        memset(head,-1,sizeof(head));
        k=0;
        int x,l;
        for(int i=2; i<=n; i++)
        {
            scanf("%d%d",&x,&l);
            addedge(i,x,l);
        }
        dfs1(1,-1);      ///随便选一点作为树根，那么他的父亲结点就设为-1
        dfs2(1,-1);       ///所以换成dfs1(2,-1),dfs2(2,-1)  也是可以的
        for(int i=1; i<=n; i++)
            printf("%d\n",fm[i]);
    }
    return 0;
}