poj 1985 Cow Marathon

原题链接：http://poj.org/problem?id=1985
树的直径。。

#include<algorithm>
#include<iostream>
#include<cstdlib>
#include<cstring>
#include<cstdio>
#include<vector>
#include<queue>
#include<map>
using std::map;
using std::min;
using std::find;
using std::pair;
using std::queue;
using std::vector;
using std::multimap;
#define pb(e) push_back(e)
#define sz(c) (int)(c).size()
#define mp(a, b) make_pair(a, b)
#define all(c) (c).begin(), (c).end()
#define iter(c) __typeof((c).begin())
#define cls(arr, val) memset(arr, val, sizeof(arr))
#define cpresent(c, e) (find(all(c), (e)) != (c).end())
#define rep(i, n) for(int i = 0; i < (int)n; i++)
#define tr(c, i) for(iter(c) i = (c).begin(); i != (c).end(); ++i)
const int N = 50010;
const int INF = 0x3f3f3f3f;
struct work {
    struct edge { int to, w, next; }G[N << 1];
    int tot, head[N], dist[N];
    inline void init() {
        tot = 0, cls(head, -1);
    }
    inline void add_edge(int u, int v, int w) {
        G[tot] = (edge){ v, w, head[u] }; head[u] = tot++;
        G[tot] = (edge){ u, w, head[v] }; head[v] = tot++;
    }
    inline int bfs(int s) {
        int id = s, max_dist = 0;
        cls(dist, -1);
        queue<int> q;
        q.push(s);
        dist[s] = 0;
        while(!q.empty()) {
            int u = q.front(); q.pop();
            if(dist[u] > max_dist) {
                max_dist = dist[id = u];
            }
            for(int i = head[u]; ~i; i = G[i].next) {
                edge &e = G[i];
                if(-1 == dist[e.to]) {
                    dist[e.to] = dist[u] + e.w;
                    q.push(e.to);
                }
            }
        }
        return id;
    }
    inline void solve(int m) {
        char ch;
        int u, v, w;
        cls(head, -1);
        while(m--) {
            scanf("%d %d %d %c", &u, &v, &w, &ch);
            add_edge(u, v, w);
        }
        printf("%d\n", dist[bfs(bfs(u))]);
    }
}go;
int main() {
#ifdef LOCAL
    freopen("in.txt", "r", stdin);
    freopen("out.txt", "w+", stdout);
#endif
    int n, m;
    while(~scanf("%d %d", &n, &m)) {
        go.solve(m);
    }
    return 0;
}