HDU - 2196（树形dp）

给出一颗树，求树中的每个顶点到其他所有顶点距离的最大值。

经典树形dp

#include <bits/stdc++.h>

using namespace std;
const int maxn=1e4+100;
int n,dp[maxn],dpf[maxn],f[maxn][3];//dp为孩子节点到这个点的最远距离，dpf为父节点到这个节点的最远距离，最后取max
struct node{
    int x,val;
    node(int _x,int _val)
    {
        x=_x;
        val=_val;
    }
};
vector<node>a[maxn];
void dfs(int u,int fa)
{
    for(int i=0;i<a[u].size();i++)
    {
        int v=a[u][i].x;
        if(v==fa) continue;
        dfs(v,u);
        dp[u]=max(dp[u],dp[v]+a[u][i].val);////
    }
}
void dfsI(int u,int fa)
{
    int tmp1=0,x=0,tmp2=0,y=0;
    for(int i=0;i<a[u].size();i++)
    {
        int v=a[u][i].x;
        if(v==fa) continue;
        if(dp[v]+a[u][i].val>tmp2)////找到兄弟节点到这个点的最远和次远距离并记录哪个兄弟（因为最远距离可能是从自己这个方向过来的）
        {
            tmp2=dp[v]+a[u][i].val;
            y=v;
            if(tmp1<tmp2)
            {
                swap(x,y);
                swap(tmp1,tmp2);
            }
        }
    }
    for(int i=0;i<a[u].size();i++)
    {
        int v=a[u][i].x;
        if(v==fa) continue;
        if(v==x)
        dpf[v]=max(dpf[u],tmp2)+a[u][i].val;//最远的正好是自己这边来的|父亲节点来的
        else
            dpf[v]=max(dpf[u],tmp1)+a[u][i].val;///最远的是兄弟节点过来的|父亲节点来的
        dfsI(v,u);
    }
}
int main()
{
   /// freopen("in.txt","r",stdin);
    while(~scanf("%d",&n))
    {
        for(int i=1;i<=n;i++)
            a[i].clear();
        for(int i=2;i<=n;i++)
        {
            int x,w;
            scanf("%d%d",&x,&w);
            a[x].push_back(node(i,w));
            a[i].push_back(node(x,w));
        }
        memset(dp,0,sizeof dp);
        memset(dpf,0,sizeof dpf);
        memset(f,0,sizeof f);
        dfs(1,-1);
        dfsI(1,-1);
        for(int i=1;i<=n;i++)
            printf("%d\n",max(dp[i],dpf[i]));
    }
    return 0;
}