Hdu 2196 Computer 经典树形dp

题目链接：点击打开链接

题意：给一棵树，每条树边都有权值，问从每个顶点出发，经过的路径权值之和最大为多少？n <= 10000,权值<=10^9
分析：我的第一道树形dp题，看了几篇博客，一般的树形dp题，差不多就是一遍或两遍dfs，维护从根到叶子或从叶子到根的状态转移。这题就是典型的树形dp题。第一遍dfs从叶子到根，统计出以每个点为根的子树的最长距离和次长距离，然后第二次dfs从根到叶子，求出每个点通过父节点得到的最长距离。为什么这样做就可以得出正解？这篇文章做出了详细解释，知道了怎么做之后，剩下的就是敲代码了。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int N=1e4+5;
int head[N];
struct edge
{
    int v,w,next;
}e[N];
int cnt;
int f[N][3]; //f[i][0]表示以节点i为根的子树的最长距离，f[i][1]是次长距离，f[i][2]是结点i通过父节点得到的最长距离。
void init()
{
    memset(head,-1,sizeof(head));
    cnt=0;
}
void add_edge(int u,int v,int w) //建立一个有向图u->v
{
    e[cnt].v=v;
    e[cnt].w=w;
    e[cnt].next=head[u];
    head[u]=cnt++;
}
void dfs1(int u)
{
    int biggest=0,bigger=0;
    for(int i=head[u];~i;i=e[i].next){
        int v=e[i].v;
        dfs1(v);
        int tmp=f[v][0]+e[i].w;
        if(biggest<=tmp){
            bigger=biggest;
            biggest=tmp;
        }
        else if(bigger<tmp){
            bigger=tmp;
        }
    }
    f[u][0]=biggest;
    f[u][1]=bigger;

}
void dfs2(int u)
{
    for(int i=head[u];~i;i=e[i].next){
        int v=e[i].v;
        f[v][2]=max(f[u][2],f[v][0]+e[i].w==f[u][0]?f[u][1]:f[u][0])+e[i].w;
        dfs2(v);
    }
}
int main()
{
    int n,u,v,w;
    //freopen("f.txt","r",stdin);
    while(~scanf("%d",&n)){
        init();
        for(int v=2;v<=n;v++){
            scanf("%d%d",&u,&w);
            add_edge(u,v,w);
        }
        dfs1(1);
        f[1][2]=0;
        dfs2(1);
        for(int i=1;i<=n;i++)
            printf("%d\n",max(f[i][0],f[i][2]));
    }
    return 0;
}