[POJ 2631/UVA 10308 Roads in the North] DFS求树上的最长路

[POJ 2631/UVA 10308 Roads in the North] DFS求树上的最长路

题目链接： [POJ 2631Roads in the North]

题意：求树上的最长路。

分析：第一次从任意节点为根节点开始DFS每个节点距离根节点的距离，那么距离最大的那个节点就必然是最长路的一个端点；然后再从这个端点为根节点出发，再次DFS其他节点距离该节点的最长距离。 这个最长距离就是答案了。

#include <cmath>
#include <queue>
#include <vector>
#include <cstdio>
#include <string>
#include <cstring>
#include <iomanip>
#include <iostream>
#include <algorithm>
using namespace std;

//#pragma comment(linker, "/STACK:1024000000,1024000000")

#define FIN             freopen("input.txt","r",stdin)
#define FOUT            freopen("output.txt","w",stdout)
#define fst             first
#define snd             second

typedef __int64  LL;
//typedef long long LL;
typedef unsigned int uint;
typedef pair<int, int> PII;

const int MAXN = 1e5 + 5;

struct Edge {
    int v, w, next;
    Edge() {}
    Edge (int v, int w, int next) : v (v), w(w), next (next) {}
}edges[MAXN << 1];
int head[MAXN];
bool vis[MAXN];
int d, id;
void init() {
    memset(head, -1, sizeof(head));
}
void add_edge(int u, int v, int w) {
    static int tot = 0;
    edges[tot] = Edge(v, w, head[u]);
    head[u] = tot++;
}
void dfs(int u, int pre, int dist) {
    int v, w;
    vis[u] = true;
    if(d < dist) {
        d = dist;
        id = u;
    }
    for(int i = head[u]; ~i; i = edges[i].next) {
        v = edges[i].v, w = edges[i].w;
        if(v == pre || vis[v]) continue;
        dfs(v, pre, dist + w);
    }
    vis[u] = false;
}
int main() {
#ifndef ONLINE_JUDGE
    FIN;
#endif // ONLINE_JUDGE
    init();
    int u, v, w;
    while(~scanf("%d %d %d", &u, &v, &w)) add_edge(u, v, w), add_edge(v, u, w);
    d = 0, id = 1;
    dfs(1, -1, 0);
    dfs(id, -1, 0);
    printf("%d\n", d);
    return 0;
}