HDU 2196 Computer（经典树形DP）

题目链接：Click here~~

题意：

给一颗树，求出每个点到其距离最远的点的距离。

解题思路：

超经典的树形DP，最近经常遇到这种题目，一定要把这道题写篇博客记录下。

考虑到树中每个点的最远距离其实只有两种情况，从子树 或 从父亲 得到。（Note：from sons or father）

首先从根节点跑一次 dfs，很容易求出 u 到子树中的最远距离和次远距离，解决了 from sons。（Note：from sons of v => from sons of u）

再从上到下对 from father 进行 dp，对于<u,v> 关于 v 的转移既可以从 u 的父亲也可以从 u 的子树得到，解决 from father。（Note：from father of v ≠> from father of u）

而维护次远距离就是为了 from sons of u 时不要取到 v。

最后再取个 max 就好了。

#include <stdio.h>
#include <string.h>
#include <algorithm>
using namespace std;

const int N = 1e4 + 5;

struct Vertex{
    int head;
}V[N];

struct Edge{
    int v,w,next;
}E[N];

int top;

void init()
{
    memset(V,-1,sizeof(V));
    top = 0;
}

void add_edge(int u,int v,int w)
{
    E[top].v = v;
    E[top].w = w;
    E[top].next = V[u].head;
    V[u].head = top++;
}

int dp[N][3];

void dfs1(int u)
{
    int biggest = 0 , bigger = 0;
    for(int i=V[u].head;~i;i=E[i].next)
    {
        int v = E[i].v;
        dfs1(v);
        int tmp = dp[v][0]+E[i].w;
        if(biggest <= tmp)
        {
            bigger = biggest;
            biggest = tmp;
        }
        else if(bigger < tmp)
            bigger = tmp;
    }
    dp[u][0] = biggest;
    dp[u][1] = bigger;
}

void dfs2(int u)
{
    for(int i=V[u].head;~i;i=E[i].next)
    {
        int v = E[i].v;
        dp[v][2] = max(dp[u][2] , dp[v][0]+E[i].w==dp[u][0] ? dp[u][1] : dp[u][0]) + E[i].w;
        dfs2(v);
    }
}

int main()
{
    int n;
    while(~scanf("%d",&n))
    {
        init();
        for(int v=2;v<=n;v++)
        {
            int u,w;
            scanf("%d%d",&u,&w);
            add_edge(u,v,w);
        }
        dfs1(1);
        dp[1][2] = 0;
        dfs2(1);
        for(int i=1;i<=n;i++)
            printf("%d\n",max(dp[i][0],dp[i][2]));
    }
	return 0;
}