（水）大臣的旅费 蓝桥杯 树的直径

题目大意：给定一个图，每2个城市都只有一条路联通，求最长路

分析：因为每2个城市都只有一条路联通所以为一颗树，要在树中求最长路即求树的直径，我们有一个求树的直径的算法，只要2次dfs（）就够了。第一次随便找个点dfs（）找到与其最远的点a，则点a必定为树的直径的起点或者终点，具体证明读者可以自己画个图用反证法，先假定并画出树的直径，然后随便找一个点并假设其最远点a不在树的直径的起点或者终点上，那么此时在树的直径中必然可以找一点v与a连接后比树的直径更长，所以与假设矛盾。第二次dfs（）找到的最远路便是答案了。

直接上代码：

#include <cstdio>               //树的直径的求法  2次dfs
#include <cstring>  
#include<vector>
#include <algorithm>  
using namespace std;
#define Max 10000+5
struct Ege
{
	int to, l;
	Ege(int a = 0, int b = 0){ to = a, l = b; }
};
vector<Ege> p[Max];
int n, ans, t;    //t用来记录最长路的终点
bool vis[Max] = { 0 };
void dfs(int cur, int sum)
{
	if (sum > ans) { ans = sum; t = cur; }   
	for (int i = 0; i < p[cur].size(); i++)
	{
		int To = p[cur][i].to, L = p[cur][i].l;
		if (!vis[To])
		{
			vis[To] = 1;
			dfs(To, sum + L);
			vis[To] = 0;
		}
	}
	return;
}
int main()
{
	scanf("%d", &n);
	for (int i = 1; i < n; i++)
	{
		int a, b, c;
		scanf("%d%d%d", &a, &b, &c);
		p[a].push_back(Ege(b, c));
		p[b].push_back(Ege(a, c));
	}
	vis[1] = 1;
	dfs(1, 0);
	vis[1] = 0;
	vis[t] = 1;
	dfs(t, 0);
	vis[t] = 0;
	printf("%d\n", ans * 10 + ans*(ans + 1) / 2);
	return 0;
}