C++-HDU2196-Computer-[树的直径]

直径定义：树上的最长路径，不妨设端点分别为s，t

可以证明（感觉）：每个点到其最远点必定为s or t，反之亦然

首先，第一次dfs找到s

然后，第二次dfs以s为根找到t

最后，第三次dfs以t为根

比较二三两次的当前点深度可得到答案

#include <set>
#include <map>
#include <cmath>
#include <queue>
#include <vector>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
const int MAXN=10010,MAXM=20010;
struct edge{
    int u,v,w,next;
    edge(int u=0,int v=0,int w=0,int next=0):u(u),v(v),w(w),next(next){}
};

edge E[MAXM];
int head[MAXN],fa[MAXN],dis[MAXN],ans[MAXN],cnt,n,s,t,maxdis;
void add(int u,int v,int w){
    E[++cnt]=edge(u,v,w,head[u]),head[u]=cnt;
    E[++cnt]=edge(v,u,w,head[v]),head[v]=cnt;
}

int dfs(int x,int fa){
    for(int i=head[x];i;i=E[i].next)
        if(E[i].v!=fa){
            dis[E[i].v]=max(dis[x]+E[i].w,dis[E[i].v]);
            dfs(E[i].v,x);
        }
}

int search(int root){
    int x=root;
    maxdis=0,memset(dis,0,sizeof(dis)),dfs(root,root);
    for(int i=1;i<=n;i++)if(dis[i]>maxdis)maxdis=dis[i],x=i;
    return x;
} 

int main(){
    while(scanf("%d",&n)!=EOF){
        cnt=0;
        memset(head,0,sizeof(head));
        for(int u=2,v,w;u<=n;u++)scanf("%d%d",&v,&w),add(u,v,w);
        int s=search(1);
        int t=search(s);
        for(int i=1;i<=n;i++)ans[i]=dis[i];
        search(t);
        for(int i=1;i<=n;i++)printf("%d
",max(ans[i],dis[i]));
    }
    return 0;
}