lightoj 1257 - Farthest Nodes in a Tree (II) 树直径

给定一棵树，问树上每个点到其他的点最远的距离。

首先所有点到树的直径两端点肯定是最远的（请百度树的直径，所以就是所有点到树的直径两端点那个更长..

任选一点跑一边搜索找出来树的直径一个端点，在一遍搜索就可以找到另外一个端点，顺便求得所有点到这个端点的距离啦。

然后再来一遍求另外一个端点到所有点的距离...

#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define ull unsigned long long
#define mod 1000000007
#define inf 0x3f3f3f3f
int dp[40000];
struct edge
{
int u,v,w,next;
}edge[80000];
int head[40000],tot,used[40000],s,mlen;
void init()
{
memset(dp,0,sizeof(dp));
memset(head,-1,sizeof(head));
memset(used,0,sizeof(used));
tot=0;
}
void add(int u,int v,int w)
{
edge[tot].v=v;
edge[tot].w=w;
edge[tot].next=head[u];
head[u]=tot++;
}
void dfs(int u,int pre,int len)
{
if(mlen<len)
{
s=u;
mlen=len;
}
for(int i=head[u];i!=-1;i=edge[i].next)
{
int v=edge[i].v;
int w=edge[i].w;
if(v==pre) continue;
dfs(v,u,len+w);
dp[v]=max(dp[v],len+w);
}
}
int main()
{
int t;
scanf("%d",&t);
for(int cas=1;cas<=t;cas++)
{
int n,u,v,w;
init();
scanf("%d",&n);
for(int i=0;i<n-1;i++)
{
scanf("%d %d %d",&u,&v,&w);
add(u,v,w);
add(v,u,w);
}
s=mlen=0;
dfs(1,-1,0);
dfs(s,-1,0);
dfs(s,-1,0);
printf("Case %d:\n",cas);
for(int i=0;i<n;i++)
printf("%d\n",dp[i]);
}
return 0;
}