树直径问题

树的直径

我们把一棵树的最长的路径称作为数的直径。
怎么求数的直径？
1、从任意一个节点出发，用bfs或者dfs遍历，找到最长路径的最后一个点，那个就是直径的一个端点a。
2、从一个端点a出发，再次使用bfs或者dfs遍历，找到最长路径的最后一个点，就是直径的另外一个端点b。
也就是要两次遍历。

具体题目：

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

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

分析

这个，最直观的思路就是求出数的直径，记录两个端点a,b，然后循环每个节点，计算这个结点到两个端点的距离，取最长的那个。
但是如果逆过来思考的话，那就是从a端点出发，遍历每个结点，然后记录a到各个结点的路径长度；然后从b端点出发，遍历每个结点，然后记录b到各个结点的路径长度。

具体代码

#include<stdio.h>
#include<queue>
#include<vector>
using namespace std;
struct point
{
	int u,v,w;
};
vector<point> G[10010];
bool visit[10010];
int n,dis[10010];
int bfs(int begin)
{
	
	for(int i=0;i<=n;i++){//初始化距离
		dis[i]=0;visit[i]=false;}
	queue<int>Q;Q.push(begin);visit[begin]=1;dis[begin]=0;
	while(!Q.empty())
	{
		int a=Q.front();
		Q.pop();
		for(int i=0;i<G[a].size();i++)//遍历所有相连的点
		{
			if(visit[G[a][i].v]==false)//没有遍历过
			{
				visit[G[a][i].v]=true;
				Q.push(G[a][i].v);
				dis[G[a][i].v]=dis[a]+G[a][i].w;
			}
		}
	}
	int ans=0,themax=0;
	for(int i=1;i<=n;i++)
		if(dis[i]>themax){
			ans=i;themax=dis[i];
		}
	return ans;
}
int main()
{
	while(scanf("%d",&n)!=EOF)
	{
		int maxdis[10010];
		for(int i=0;i<=n;i++)   G[i].clear();
		for(int i=2;i<=n;i++)//将图的各边和权重输入
		{
			int v,w;scanf("%d %d",&v,&w);
			point p;
			p.u=i;p.v=v;p.w=w;//无向图
			G[i].push_back(p);
		    	p.u=v;p.v=i;p.w=w;
			G[v].push_back(p);
		}
		//寻找树的直径的两个端点
		int v1=bfs(1);//第一个端点
		//从第一个端点开始遍历这个图
		int v2=bfs(v1);//第二个端点
		for(int i=1;i<=n;i++) maxdis[i]=dis[i];//到第一个端点的距离
		bfs(v2);//从第二个端点开始遍历这个图
		for(int i=1;i<=n;i++){
			if(dis[i]>maxdis[i]){
				maxdis[i]=dis[i];
			}
			printf("%d\n",maxdis[i]);
		}
	}
	return 0;
}