买房子2

题目描述
有天小C突发奇想，自己是不是也该考虑一下买房子的问题了。小C所在的城市被划分成n个区域，这n个区域是连通的，并且从任意一个区域到达另外区域的方案数只有一种。现在这n个区域都有房卖，小C想，如果他要选择买房区域的话，他所在的区域到其他的区域的距离的最远距离是多少。现在告诉你n个区域的连接情况，请你帮他算算，输出对于每个区域离它最远的距离是多少？
输入 输入第一行，一个整数n; 接下来n-1行，每行三个整数a,b,c，表示连接区域a和b的路长为c，其中0<=a,b< n,0< c <=10000。
输出 输出对于每个区域离它最远的距离是多少？

一道经典的求任一任意点最远距离的题目 复杂度O（n）

#include<cstdio>
#include<vector>
using namespace std;
#define FOR(i,x,y) for(int i=(x);i<(y);i++)//左闭右开 
#define M 20005
#define oo 1000000000
int n;
//三遍dfs找到每个点的对应最远点
struct node {int id,v;};
vector<node>edge[M];
int dis[M];
int mxlen=0,dx;
void dfs(int x,int f,int d) {
    if(dis[x]<d)dis[x]=d;
    if(mxlen<d){dx=x;mxlen=d;}
    FOR(i,0,edge[x].size()) {
        int y=edge[x][i].id;
        if(y==f)continue;
        dfs(y,x,d+edge[x][i].v);
    }
}
int main() {
    scanf("%d",&n);
    FOR(i,1,n) {
        int a,b,c;
        scanf("%d %d %d",&a,&b,&c);
        edge[a].push_back((node) {b,c});
        edge[b].push_back((node) {a,c});
    }
    dfs(0,-1,0);dfs(dx,-1,0);dfs(dx,-1,0);
    FOR(i,0,n)printf("%d\n",dis[i]);
    return 0;
}