HDU 2196 Computer (树形DP必做）

【题意】求树上每个点的最远距离是多少

这题一直被称为树形dp的经典是有它的道理的，因为树dp就是把dp放到树上做了，一般是从上到下或从下到上（利用回溯）的移转状态。而这题很合适的需要两次dfs。

对于<u,v>（有向），

dp[u][0]表示在u的子树下u的最远距离是多少

dp[u][1]表示在u的子树下（和dp[u][0]不是同一孩子）u的次远距离是多少

dp[u][2]表示通过u的父亲能走的最远距离是多少

第一次从下到上，对于<u,v>（有向），状态转移显然是 dp[u][0] = dp[v][0]+w[i]；所以要先算出dp[v][0]才能知道dp[u][0]。故是从下往上。

第二次从上往下，其实就是再遍历一边图，把dp[v][2]算出来，显然：

dp[v][2] = max(dp[u][2]，dp[v][0]+w[i]==dp[u][0]?dp[u][1]:dp[u][0]) + w[i]；

要算dp[v][2]，要先算dp[u][0]，所以从上往下。

最后的答案就是 max（dp[u][0]，dp[u][2]）

【代码】

<span style="font-size:14px;">/* ***********************************************
Author        :angon

************************************************ */
#include <stdio.h>
#include <string.h>
#include <iostream>
#include <algorithm>
#include <stack>
#include <vector>
#include <queue>
#include <set>
#include <map>
#include <string>
#include <math.h>
#include <stdlib.h>
#include <time.h>
using namespace std;
#define showtime fprintf(stderr,"time = %.15f\n",clock() / (double)CLOCKS_PER_SEC)
#define lld %I64d
#define REP(i,k,n) for(int i=k;i<n;i++)
#define REPP(i,k,n) for(int i=k;i<=n;i++)
#define scan(d) scanf("%d",&d)
#define scanl(d) scanf("%I64d",&d)
#define scann(n,m) scanf("%d%d",&n,&m)
#define scannl(n,m) scanf("%I64d%I64d",&n,&m)
#define mst(a,k)  memset(a,k,sizeof(a))
#define LL long long
#define N 10005
#define mod 1000000007
inline int read(){int s=0;char ch=getchar();for(; ch<'0'||ch>'9'; ch=getchar());for(; ch>='0'&&ch<='9'; ch=getchar())s=s*10+ch-'0';return s;}

struct Edge
{
    int v,w,next;
}edge[N*2];
int head[N],tot;
void addedge(int u,int v,int w)
{
    edge[tot].v=v;
    edge[tot].w=w;
    edge[tot].next=head[u];
    head[u]=tot++;
}
int dp[N][3];

void dfs1(int u,int v) //每个节点子树下的最大和次大
{
    for(int i=head[u];i!=-1;i=edge[i].next)
    {
        int v=edge[i].v;
        dfs1(v,u);
        int w=edge[i].w;
        int temp = dp[v][0] + w;
        if(temp >= dp[u][0])
        {
            dp[u][1]=dp[u][0];
            dp[u][0] = temp;
        }
        else if(temp > dp[u][1])
            dp[u][1] = temp;
    }
    //printf("u=%d %d %d\n",u,dp[u][0],dp[u][1]);
}

void dfs2(int u,int v)
{
    for(int i=head[u];i!=-1;i=edge[i].next)
    {
        int v=edge[i].v;
        if(dp[u][0] == dp[v][0] + edge[i].w)
        {
            dp[v][2] = max(dp[u][2],dp[u][1]) + edge[i].w;
        }
        else
        {
            dp[v][2] = max(dp[u][2],dp[u][0]) + edge[i].w;
        }
        dfs2(v,u);
    }
}
int main()
{
    //freopen("in.txt","r",stdin);
    //freopen("out.txt","w",stdout);
    int n;
    while(~scan(n))
    {
        mst(head,-1); tot=0;
        for(int v=2;v<=n;v++)
        {
            int u,w; scann(u,w);
            addedge(u,v,w);
        }
        mst(dp,0);
        dfs1(1,-1);
        dfs2(1,-1);
        for(int i=1;i<=n;i++)
            printf("%d\n",max(dp[i][0],dp[i][2]));
    }
    return 0;
}</span>