题目链接http://acm.hdu.edu.cn/showproblem.php?pid=2196
题意:
在一个树中,求各个点到其他点的最远距离。
思路:
我们可以使用搜索求出一个点到其他点的最远距离,因此很容易想到对每个点进行一次搜索,但这个复杂度就比较高了,会超时。树上直径有一个性质:假设 a a 和是树上直径的两端点,那么对于其他任一点 i i 到其他点的最远距离。通过这个性质我们可以通过3次搜索就求出答案来。第一次,从任一点 i i 开始搜索,找到距离点最远的点不妨设为 a a ,然后从开始搜索,此时将 a a 到各点的最远距离记录在数组中,并找到距离 a a 最远的点,不妨设为,然后从 b b 开始搜索,将到各点的距离记录到数组 d2 d 2 中。最后 d[i]=max(d1[i],d2[i]),0≤i<n d [ i ] = m a x ( d 1 [ i ] , d 2 [ i ] ) , 0 ≤ 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;
}