hdu 2196 Computer（树形DP）

题意：题目给出电脑连接的方式，求出每一台电脑和离它最远的电脑的距离。

解法：把图画出来，距离一个节点最长的距离有两种可能：第一种是从它的子节点到达该点的距离，第二种是通过它的父节点到达该点的距离。第二种情况分两种：如果该点不在其父节点的最长路径上，直接用其父节点的最长路径加上其父节点与该节点之间的距离。如果该点在其父节点的最长路径上，则用其父节点的次长路径加上该点到其父节点的距离。

刚开始学树形DP。

#include<stdio.h>
#include<string.h>
#define N 10005
struct node
{
    int x,y;
    int len;
    int next;
} map[N*2];
struct point
{
    int max;
    int secMax;
} a[N];
int head[N],mark[N];
int cnt;
void addEdge(int x,int y,int k)
{
    map[cnt].x=x;
    map[cnt].y=y;
    map[cnt].len=k;
    map[cnt].next=head[x];
    head[x]=cnt++;
    map[cnt].x=y;
    map[cnt].y=x;
    map[cnt].len=k;
    map[cnt].next=head[y];
    head[y]=cnt++;
    return ;
}
int dfs(int f)
{
    int u=head[f];
    mark[f]=1;
    while(u!=-1)
    {
        int k=map[u].y;
        if(!mark[k])
        {
            int temp=dfs(k)+map[u].len;
            if(temp>a[f].secMax) a[f].secMax=temp;
            if(a[f].secMax>a[f].max) temp=a[f].max,a[f].max=a[f].secMax,a[f].secMax=temp;
        }
        u=map[u].next;
    }
    return a[f].max;
}
void DFS(int f)
{
    int u=head[f];
    mark[f]=1;
    while(u!=-1)
    {
        int k=map[u].y;
        if(!mark[k])
        {
            if(a[f].max!=a[k].max+map[u].len)
            {
                a[k].secMax=a[k].max;
                a[k].max=a[f].max+map[u].len;
            }
            else
            {
                if(a[k].max<a[f].secMax+map[u].len)
                {
                    a[k].secMax=a[k].max;
                    a[k].max=a[f].secMax+map[u].len;
                }
                else if(a[k].secMax<a[f].secMax+map[u].len)
                    a[k].secMax=a[f].secMax+map[u].len;
            }
            DFS(k);
        }
        u=map[u].next;
    }
    return ;
}
int main()
{
    int n;
    while(scanf("%d",&n)!=EOF)
    {
        memset(head,-1,sizeof(head));
        cnt=0;
        for(int i=2; i<=n; i++)
        {
            int x,k;
            scanf("%d%d",&x,&k);
            addEdge(i,x,k);
        }
        memset(a,0,sizeof(a));
        memset(mark,0,sizeof(mark));
        dfs(1);
        memset(mark,0,sizeof(mark));
        DFS(1);
        for(int i=1; i<=n; i++)
            printf("%d\n",a[i].max);
    }
    return 0;
}