氪金带东（每个节点在图中能到达的最远距离）----链式前向星&树的直径

问题描述

实验室里原先有一台电脑(编号为1)，最近氪金带师咕咕东又为实验室购置了
N-1台电脑，编号为2到N。每台电脑都用网线连接到一台先前安装的电脑上。但是咕咕东担心网速太慢，他希望知道第i台电脑到其他电脑的最大网线长度，但是可怜的咕咕东在不久前刚刚遭受了宇宙射线的降智打击，请你帮帮他。

提示: 样例输入对应这个图，从这个图中你可以看出，距离1号电脑最远的电脑是4号电脑，他们之间的距离是3。 4号电脑与5号电脑都是距离2号电脑最远的点，故其答案是2。5号电脑距离3号电脑最远，故对于3号电脑来说它的答案是3。同样的我们可以计算出4号电脑和5号电脑的答案是4.

Input

输入文件包含多组测试数据。对于每组测试数据，第一行一个整数N (N<=10000)，接下来有N-1行，每一行两个数，对于第i行的两个数，它们表示与i号电脑连接的电脑编号以及它们之间网线的长度。网线的总长度不会超过10^9，每个数之间用一个空格隔开。

Output

对于每组测试数据输出N行，第i行表示i号电脑的答案 (1<=i<=N)

Sample Input

5
1 1
2 1
3 1
1 1

Sample Output

3
2
3
4
4

解题思路以及关键代码

关于链式前向星–参考博客 深度理解链式前向星

//链式前向星 
struct Edge
{
	int next;
	int to;
	int w;
}edge[2*maxn];
int head[maxn],cnt;
void init()
{
	cnt = 0;
	memset(head,-1,sizeof(head));
}
void add(int u,int v,int w)
{
	edge[cnt].to=v;
	edge[cnt].w =w;
	edge[cnt].next = head[u];
	head[u] = cnt;
	cnt++;
}

①如果对所有点进行一次bfs遍历，根据数据范围可知O(n^2)，显然会TLE

for (int i=1; i<=n; i++)
     printf("%d\n",bfs(i));//bfs返回点能到达的最远距离

②利用树的直径。
首先回顾求解树的直径算法

详细证明参考树的直径算法正确性数学证明

a = bfs(1); //直径左端点 
b = bfs(a); //直径右端点
for (int i=1; i<=n; i++)
    tmp[i]=dis[i];
bfs(b);
for (int i=1; i<=n; i++)
    printf("%d\n",max(tmp[i],dis[i]));

由树的直径算法第一点可知 任意点到它最远距离的点一定是直径的一端
那么可以反过来，先找到直径的两端，然后从两边分别进行一次遍历，然后过程中记录路径长。最终每个节点等于两次遍历结果的max。

全部代码

#include<iostream>
#include<cstdio>
#include<cstring>
#include<queue> 
using namespace std;
const int maxn = 1e4 + 5;
//链式前向星 
struct Edge
{
	int next;
	int to;
	int w;
}edge[2*maxn];
int head[maxn],cnt;
void init()
{
	cnt = 0;
	memset(head,-1,sizeof(head));
}
void add(int u,int v,int w)
{
	edge[cnt].to=v;
	edge[cnt].w =w;
	edge[cnt].next = head[u];
	head[u] = cnt;
	cnt++;
}
//bfs 
bool vis[maxn];
int dis[maxn],tmp[maxn],node,a,b,sum;
int bfs(int begin)
{
    queue<int> q;
    //初始化 
    memset(dis,0,sizeof(dis));
    memset(vis,0,sizeof(vis));
    while(!q.empty()) 
		q.pop();
	//原点 
    vis[begin]=true;
    q.push(begin);
    node=begin;
    sum=0;

    while(!q.empty())
    {
        int now=q.front(); q.pop();
        for (int i=head[now]; ~i; i=edge[i].next) //-1的补码是全1  ~按位取反  ~（-1）= 0 即条件为假 
        {
            if(!vis[edge[i].to])
            {
				vis[edge[i].to]=1;
                dis[edge[i].to]=dis[now]+edge[i].w;
                if(sum < dis[edge[i].to])//找到最远的点
                {
                    sum=dis[edge[i].to];
                    node=edge[i].to;
                }
				q.push(edge[i].to);
            }
        }
    }
    return node; //返回该点 经过距离 最远的点 
}

int main()
{
	int n,v,w;
	while(~scanf("%d",&n))
	{
		init();
		for(int i=2;i<=n;i++)
		{
			scanf("%d%d",&v,&w);
			add(i,v,w);
			add(v,i,w);
		}
		a = bfs(1); //直径左端点 
        b = bfs(a); //直径右端点
        for (int i=1; i<=n; i++)
            tmp[i]=dis[i];
        bfs(b);
        for (int i=1; i<=n; i++)
            printf("%d\n",max(tmp[i],dis[i]));
	}
	return 0;
 }