[Poj] Roads in the North

http://poj.org/problem?id=2631

树的直径裸题

dfs/bfs均可

/*
    dfs
*/
#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cmath>
#include <cstring>
#include <string>

using namespace std;
const int N = 1e5 + ; 

#define yxy getchar()

int now = , point, Max_dis;
int dis[N], head[N];
bool vis[N];
struct Node {int v, w, nxt;} G[N << ];

inline void add(int u, int v, int w){
    G[now].v = v; G[now].w = w; G[now].nxt = head[u]; head[u] = now ++;
}

void dfs(int u, int dist){
    for(int i = head[u]; ~ i; i = G[i].nxt){
        int v = G[i].v;
        if(!vis[v]){
            dis[v] = dist + G[i].w;
            vis[v] = ;
            if(dis[v] > Max_dis){
                Max_dis = dis[v];
                point = v;
            }
            dfs(v, dis[v]);
        }
    }
}

int main()
{
    memset(head, -, sizeof(head));
    int u_, v_, w_;
    while(scanf("%d%d%d", &u_, &v_, &w_) == ){
        add(u_, v_, w_); add(v_, u_, w_);
    }
    vis[] = ;
    dfs(, );
    Max_dis = ;
    memset(vis, , sizeof(vis));
    vis[point] = ;
    dfs(point, );
    cout << Max_dis;
    return ;
}

/*
    bfs
*/
#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cmath>
#include <cstring>
#include <string>
#include <queue>

using namespace std;
const int N = 1e5 + ; 

#define yxy getchar()

int now = , point, Max_dis;
int dis[N], head[N];
bool vis[N];
struct Node {int v, w, nxt;} G[N << ];
queue <int> Q;

inline void add(int u, int v, int w){
    G[now].v = v; G[now].w = w; G[now].nxt = head[u]; head[u] = now ++;
}

void bfs(int S){
    Q.push(S);
    vis[S] = ;
    while(!Q.empty()){
        int topp = Q.front();
        Q.pop();
        for(int i = head[topp]; ~ i; i = G[i].nxt){
            int v = G[i].v;
            if(!vis[v]){
                vis[v] = ;
                dis[v] = dis[topp] + G[i].w;
                Q.push(v);
                if(dis[v] > Max_dis){
                    Max_dis = dis[v];
                    point = v;
                }
            }
        }
    }
}

int main()
{
    memset(head, -, sizeof(head));
    int u_, v_, w_;
    while(scanf("%d%d%d", &u_, &v_, &w_) == ){
        add(u_, v_, w_); add(v_, u_, w_);
    }
    bfs();
    Max_dis = ;
    memset(vis, , sizeof(vis));
    memset(dis, , sizeof(dis));
    bfs(point);
    cout << Max_dis;
    return ;
}