Computer HDU - 2196 树的直径

题目链接http://acm.hdu.edu.cn/showproblem.php?pid=2196

题意：

在一个树中，求各个点到其他点的最远距离。

思路：

我们可以使用搜索求出一个点到其他点的最远距离，因此很容易想到对每个点进行一次搜索，但这个复杂度就比较高了，会超时。树上直径有一个性质：假设$a$和$b$是树上直径的两端点，那么对于其他任一点$i$到其他点的最远距离$d[i] = max(dis(i,a),dis(i,b))$。通过这个性质我们可以通过3次搜索就求出答案来。第一次，从任一点$i$开始搜索，找到距离$i$点最远的点不妨设为$a$,然后从$a$开始搜索，此时将$a$到各点的最远距离记录在$d1$数组中，并找到距离$a$最远的点，不妨设为$b$,然后从$b$开始搜索，将$b$到各点的距离记录到数组$d2$中。最后$d[i] = max(d1[i],d2[i]),0\le i <n$就是所求结果。

代码：

#include <bits/stdc++.h>
#define N 10005
#define FP(PP,QQ,RR) for(int PP = QQ;PP < RR;PP ++)
using namespace std;
template <class T>
inline void in(T &ret) 
{
    char c; 
    ret = 0;
    while ((c = getchar()) < '0' || c > '9');
    while (c >= '0' && c <= '9')
    { 
        ret = ret * 10 + (c - '0'), c = getchar();
    }
}
struct node{
    int to,cost;
};
vector <node> G[N];
int d1[N],d2[N],n;
bool used[N];
queue <int> q;
void add(int u,int v,int c){
    G[u].push_back({v,c});
    G[v].push_back({u,c});
}
void dis(int * d){
    FP(i,0,n){
        cout << d[i] << " \n"[i + 1 == n];
    }
}
void bfs(int s,int * d){
    memset(d,0,sizeof(int) * n);
    memset(used,0,sizeof used);
    used[s] = 1;
    q.push(s);
    while(!q.empty()){
        s = q.front();q.pop();
        FP(i,0,G[s].size()){
            int u = G[s][i].to;
            if(!used[u]){
                d[u] = d[s] + G[s][i].cost;
                used[u] = 1;
                q.push(u);
            }
        }
    }
}
int maxi(int * d){
    return max_element(d,d + n) - d;
}
int main(void)
{
    while(cin >> n){
        FP(i,1,n){
            int a,b;
            in(a);in(b);
            // scanf("%d%d",&a,&b);
            a --;
            add(i,a,b);
        }
        bfs(0,d1);
        bfs(maxi(d1),d1);
        bfs(maxi(d1),d2);
        FP(i,0,n){
            printf("%d\n",max(d1[i],d2[i]));
        }
        FP(i,0,n){
            G[i].clear();   
        }
    }
    return 0;
}