树的直径 【总结】



树的直径问题：

有N个点以及M条路径，给出M条路的长度，让你找到一条 两点(任意的)间的路径，使得距离最长，并输出最长距离。

思路：第一次BFS找最长路径的一个端点，第二次BFS求最长距离。

模板：

#include <cstdio>
#include <cstring>
#include <queue>
#include <algorithm>
#define MAXN 40000+10
#define MAXM 80000+10
#define INF 100000000
using namespace std;
struct Edge
{
	int to, val, next;
}edge[MAXM];
int dist[MAXN];//存储最长路
bool vis[MAXN];
int head[MAXN], top;
int N, M;//点 边 
int ans;//最终答案 
int node;//找寻 最长路 s-t的端点 
void init()
{
	top = 0;
	memset(head, -1, sizeof(head));
}
void addEdge(int u, int v, int w)
{
	Edge E = {v, w, head[u]};
	edge[top] = E;
	head[u] = top++;
}
void getMap()
{
	int a, b, c;
	while(M--)
	{
		scanf("%d%d%d%s", &a, &b, &c);
		addEdge(a, b, c);
		addEdge(b, a, c);
	} 
}
void BFS(int sx)
{
	queue<int> Q;
	memset(dist, 0, sizeof(dist));
	memset(vis, 0, sizeof(vis));
	node = sx;
	ans = 0;
	vis[sx] = 1;
	Q.push(sx);
	while(!Q.empty())//查询 最长路 这里可不是SPFA算法 
	{
		int u = Q.front();
		Q.pop(); 
		for(int i = head[u]; i != -1; i = edge[i].next)//遍历所有边 每个点一次就够了  
		{
			Edge E = edge[i];
			if(!vis[E.to])
			{
				dist[E.to] = dist[u] + E.val;
				if(ans < dist[E.to])//更新
				{
					ans = dist[E.to];
					node = E.to;
				}
			    vis[E.to] = 1;
				Q.push(E.to);
			} 
		}
	}
}
int main()
{
	while(scanf("%d%d", &N, &M) != EOF)
	{
		init();
		getMap();
		BFS(1);//找 最长路 S-T 端点 
		BFS(node);//从端点查询 最长路 
		printf("%d\n", ans);
	}
	return 0;
}