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=,MAXM=;
 struct edge{
     int u,v,w,next;
     edge(int u=,int v=,int w=,int next=):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=,memset(dis,,sizeof(dis)),dfs(root,root);
     for(int i=;i<=n;i++)if(dis[i]>maxdis)maxdis=dis[i],x=i;
     return x;
 } 

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