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<cstdio>#include<iostream>#include<cstring>#include<queue>using namespace std;#define maxn 111111struct 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;}