HDU 2196 树的直径应用

思路：

在dfs的过程中，记录u,v之间的距离

第一遍dfs，找到直径的端点1

第二遍dfs，找到端点2,并保存端点1到各点的距离

第三遍dfs，保存端点2到各点的距离

各点能到达的最远距离，就是分别到两个端点的距离的最大值

代码：

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn = 10008;
const int inf=0x3f3f3f3f;
int n,dis1[maxn],dis2[maxn];///距离数组
int head[maxn],cnt;///head->是总边数中的第几边，cnt->总边数
int maxv=0;
inline int read() {
    int X=0,w=1; char c=getchar();
    while (c<'0'||c>'9') { if (c=='-') w=-1; c=getchar(); }
    while (c>='0'&&c<='9') X=(X<<3)+(X<<1)+c-'0',c=getchar();
    return X*w;
}

struct edge
{
    int u,v,w,ne;
}ed[maxn<<1];

void init()
{
    cnt=0;
    maxv=0;
    for(int i=0;i<=n;i++)
        head[i]=-1,dis1[i]=0,dis2[i]=0;///初始化,dis初始值为0
}
void addedge(int u,int v,int w)
{
    ed[cnt].u=u;
    ed[cnt].v=v;
    ed[cnt].w=w;
    ed[cnt].ne=head[u];///对于u点，上一条边的编号
    head[u]=cnt++;///这条边的编号,更新head
}
void dfs(int u,int f,int c,int dis[])
{
    dis[u]=c;
    if(dis[maxv]<dis[u])maxv=u;
    for(int i=head[u];i!=-1;i=ed[i].ne)
    {
        int v=ed[i].v,w=ed[i].w;
        if(v==f)continue;///父节点已经访问过
        dfs(v,u,c+w,dis);
    }
}
int main()
{

	int u,v,w,ans,l;
    while(cin>>n)
    {
	    init();
	    for(int i=2;i<=n;i++)
        {
            v=read();
            w=read();
            addedge(i,v,w);
            addedge(v,i,w);
        }
        dis1[maxv]=0;
        dfs(1,-1,0,dis1);///找一个端点
        l=maxv,dis1[maxv]=0;
        dfs(l,-1,0,dis1);///端点1到各点的距离，找到另一个端点2
        l=maxv,dis2[maxv]=0;
        dfs(l,-1,0,dis2);///端点2到各点的距离
        for(int i=1;i<=n;i++)
        {
            ans=max(dis1[i],dis2[i]);
            cout<<ans<<endl;
        }
    }
	return 0;
}