POJ 1985 Cow Marathon 树的直径

链接：http://poj.org/problem?id=1985

题意：题目阐述不是很清楚，是一棵严格树，不存在环，求其中两点间距离最长一处。

思路：两点间距离最长即为树的直径。易得，从任意点开始DFS找到距离最长一点一定是距离最长两点之一，再从找到的点再DFS一次就可以找到数的直径。

代码：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<map>
#include<queue>
#include<stack>
#include<vector>
#include<ctype.h>
#include<algorithm>
#include<string>
#define PI acos(-1.0)
#define maxn 50005
#define maxm 100005
#define INF 1<<25
#define MAX 0x7fffffff
typedef long long ll;
using namespace std;
struct Edge
{
int v,w;
int next;
}edge[maxm];
int tot,tt;
int point[maxn];
int vv[maxn],dis[maxn];
int top=1,d;
int init()
{
top=1;
memset(edge,0,sizeof(edge));
memset(point,0,sizeof(point));
memset(vv,0,sizeof(vv));
memset(dis,0,sizeof(dis));
d=0;
}
int add_edge(int u,int v,int w)
{
edge[top].v=v;
edge[top].w=w;
edge[top].next=point[u];
point[u]=top++;
edge[top].v=u;
edge[top].w=w;
edge[top].next=point[v];
point[v]=top++;
return 0;
}
int dfs(int x)
{
vv[x]=1;
for(int i=point[x];i!=0;i=edge[i].next)
{

if(!vv[edge[i].v])
{
d+=edge[i].w;
dis[edge[i].v]=d;
dfs(edge[i].v);
d-=edge[i].w;
}
}
}
int main()
{
scanf("%d%d",&tot,&tt);
init();
for(int i=0;i<tt;i++)
{
int x,y,z;
char ss[3];
scanf("%d%d%d%s",&x,&y,&z,ss);
add_edge(x,y,z);
}
dfs(1);
int aa=0,bb;
for(int i=1;i<=tot;i++)
{
if(dis[i]>aa)
{
bb=i;
aa=dis[i];
}
}
memset(dis,0,sizeof(dis));
memset(vv,0,sizeof(vv));
dfs(bb);
aa=0;
d=0;
for(int i=1;i<=tot;i++)
{
if(dis[i]>aa)
{
bb=i;
aa=dis[i];
}
}
printf("%d\n",aa);
}