poj1985 Cow Marathon

给出一棵树，求出树中的最长连。。。

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
#define maxn 40005
struct{
    int y,c,next;
}ee[maxn<<1];
int link[maxn],t,root1,k;
void insert(int a,int b,int c)
{
    ee[++t].y=b;
    ee[t].c=c;
    ee[t].next=link[a];
    link[a]=t;
}
void dfs(int root,int father,int num)
{
    int flag=0;
    for(int i=link[root];i;i=ee[i].next)
    if(ee[i].y!=father){
        flag=1;
        dfs(ee[i].y,root,num+ee[i].c);
    }
    if(!flag&&k<num){
        k=num;
        root1=root;
    }
}
int main()
{
    int n,m;
    while(~scanf("%d%d",&n,&m)){
        t=0;
        memset(link,0,sizeof(link));
        while(m--){
            int a,b,c;
            char s[5];
            scanf("%d%d%d%s",&a,&b,&c,s);
            insert(a,b,c);
            insert(b,a,c);
        }
        k=0;
        dfs(1,1,0);
        k=0;
        dfs(root1,root1,0);
        printf("%d
",k);
    }
    return 0;
}