poj 1985 Cow Marathon 树的直径

题意:

求树的直径

分析：

dfs2次即可

ACcode

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cmath>
#include <cstring>
#define maxn 150005
#define inf 0x3f3f3f3f
#include <cstring>
#define ll long long
using namespace std;
struct N{
int to,next,w;
}my[maxn];
int head[maxn],tot,ans,ed;
int dp[maxn];
void init(){
memset(head,-1,sizeof(head));
memset(dp,0,sizeof(dp));
tot=0;
}
void add(int u,int v,int w){
my[tot].to=v;my[tot].next=head[u];
my[tot].w=w;head[u]=tot++;
}
void dfs(int st,int w){
dp[st]=1;
if(ans<w)ans=w,ed=st;
for(int i=head[st];i!=-1;i=my[i].next){
int v=my[i].to;
if(!dp[v])dfs(v,w+my[i].w);
}
}
int main(){
int n,m;
while(scanf("%d%d",&n,&m)!=EOF){
init();
int a,b,c;
char s[2];
for(int i=1;i<=m;++i){
scanf("%d%d%d%s",&a,&b,&c,s);
add(a,b,c);
add(b,a,c);
}
ans=0;
dfs(1,0);
memset(dp,0,sizeof(dp));
dfs(ed,0);
printf("%d\n",ans);
}
return 0;
}