hdu 2196 Computer(树形DP)

题目链接：hdu 2196 Computer

题意：

给你一棵n个节点,n-1条边的树，每条边有一个长度，求没，每个点到最远距离的点的长度。

题解：

紫书上也有讲：

求一棵树最长的边的方法：随便找个点，dfs一遍求到长度最长的点s。

然后以s点为根dfs一遍，同样找到长度最长的点t。

然后在以t点为根，dfs一遍,求到所有点到t点的距离，

对于每个点到最远距离的点的长度就是max(dis_s[i],dis_t[i])。

#include<bits/stdc++.h>
#define F(i,a,b) for(int i=a;i<=b;++i)
using namespace std;
const int N=1e4+7;

int g[N],v[N*2],nxt[2*N],w[2*N],ed,n,s,t,dt[N],ds[N];

void adg(int x,int y,int z){v[++ed]=y,w[ed]=z,nxt[ed]=g[x],g[x]=ed;}

void dfs(int x,int fa,int dep,int *dis)
{
    dis[x]=dep;
    for(int i=g[x];i;i=nxt[i])
        if(v[i]!=fa)dfs(v[i],x,dep+w[i],dis);
}

int main(){
    while(~scanf("%d",&n))
    {
        F(i,1,n)g[i]=0;ed=0;
        F(i,2,n)
        {
            int a,b;
            scanf("%d%d",&a,&b);
            adg(i,a,b),adg(a,i,b);
        }
        dfs(1,0,0,ds);
        int mx=0;
        F(i,1,n)if(ds[i]>mx)s=i,mx=ds[i];
        dfs(s,0,0,ds);
        mx=0;
        F(i,1,n)if(ds[i]>mx)t=i,mx=ds[i];
        dfs(t,0,0,dt);
        F(i,1,n)printf("%d\n",max(ds[i],dt[i]));
    }
    return 0;
}