POJ 1985 Cow Marathon（树的直径）

Description
给出一棵无向树，求树的直径
Input
第一行为两个整数n和m分别表示点数和边数，之后m行每行表示树的一条边
Output
输出树的直径长度
Sample Input
7 6
1 6 13 E
6 3 9 E
3 5 7 S
4 1 3 N
2 4 20 W
4 7 2 S
Sample Output
52
Solution
树的直径裸题
Code

#include
#include
#include
#include
using namespace std;
#define maxn 111111
struct node
{
    int to,c,next;
}edge[2*maxn];
int n,m,head[maxn],tot,dis[maxn],vis[maxn];
void init()
{
    memset(head,-1,sizeof(head));
    tot=0;
}
void add(int u,int v,int c)
{
    edge[tot].to=v;
    edge[tot].c=c;
    edge[tot].next=head[u];
    head[u]=tot++;
}
void dfs(int u)
{
    for(int i=head[u];~i;i=edge[i].next)
    {
        int v=edge[i].to,c=edge[i].c;
        if(!vis[v])
        {
            vis[v]=1;
            dis[v]=dis[u]+c;
            dfs(v);
        }
    }
}
int main()
{
    int u,v,c,s;char op[3];
    while(~scanf("%d%d",&n,&m))
    {
        init();
        while(m--)
        {
            scanf("%d%d%d%s",&u,&v,&c,op);
            add(u,v,c),add(v,u,c);
        }
        memset(vis,0,sizeof(vis));
        dis[1]=0,vis[1]=1;
        dfs(1);
        for(int i=2,temp=-1;i<=n;i++)
            if(dis[i]>temp)temp=dis[i],s=i;
        memset(vis,0,sizeof(vis));
        dis[s]=0,vis[s]=1;
        dfs(s);
        int ans=-1;
        for(int i=1;i<=n;i++)
            ans=max(ans,dis[i]);
        printf("%d\n",ans);
    }
    return 0;
}