搜索+数的直径：大臣的旅费

题目描述

很久以前，T 王国空前繁荣。为了更好地管理国家，王国修建了大量的快速路，用于连接首都和王国内的各大城市。

为节省经费，T 国的大臣们经过思考，制定了一套优秀的修建方案，使得任何一个大城市都能从首都直接或者通过其他大城市间接到达。同时，如果不重复经过大城市，从首都到达每个大城市的方案都是唯一的。

J 是 T 国重要大臣，他巡查于各大城市之间，体察民情。所以，从一个城市马不停蹄地到另一个城市成了 J 最常做的事情。他有一个钱袋，用于存放往来城市间的路费。

聪明的 J 发现，如果不在某个城市停下来修整，在连续行进过程中，他所花的路费与他已走过的距离有关，在走第 x 千米到第 x +1 千米这一千米中（ x 是整数），他花费的路费是 x +10 这么多。也就是说走 1 千米花费 11，走 2 千米要花费 23。

J 大臣想知道：他从某一个城市出发，中间不休息，到达另一个城市，所有可能花费的路费中最多是多少呢？

输入描述

输入的第一行包含一个整数 n，表示包括首都在内的 T 王国的城市数。

城市从 1 开始依次编号，1 号城市为首都。

接下来 n -1 行，描述 T 国的高速路（ T 国的高速路一定是 n -1 条）。

每行三个整数 Pi​,Qi​,Di​，表示城市 Pi​ 和城市 Qi​ 之间有一条高速路，长度为 Di​ 千米。

输出描述:

输出一个整数，表示大臣 J 最多花费的路费是多少。

输入输出样例

示例

输入

5
1 2 2
1 3 1
2 4 5
2 5 4
 

输出：

135

【样例说明】

大臣 J 从城市 4 到城市 5 要花费 135 的路费。

运行限制

• 最大运行时间：5s
• 最大运行内存: 64M

思路分析：

  首先城市与城市之间的路可以抽象成一个有权的树结构。因为题目确保了唯一路径，所以不可能存在环。

  那我们需要做的就是求出这个树结构中两个结点之间的最长距离：即树的直径！

如何来考虑树的直径呢？

 步骤·：

1.任意取树中一点st，去遍历搜索（bfs）距离它遥远的点end

2.再从点end出发，去遍历搜索距离它最远的点，得到的长度（dist【蓝色】）就是数的长度

3.求出数的直径后，我们再根据数学公式推导出大臣的花费

即：d*10+d*（d+1）/2 

代码展示：

#include <iostream>
#include<vector>
using namespace std;
const int N =100010;

int n;
struct Edge
{
  int id,w;
};
vector<Edge> h[N];
int dist[N];

void dfs(int u,int father,int distance)
{
  dist[u] = distance;

  for(auto node : h[u])   //循环遍历的另一种表示方法 
    if(node.id!= father)   //只有不往回走才去dfs
      dfs(node.id,u,distance + node.w);
}

int main()
{
  scanf("%d",&n);
  for(int i=0;i<n-1;i++)
  {
    int a,b,c;
    scanf("%d%d%d",&a,&b,&c);
    h[a].push_back({b,c});  //与a相连的b的价值是c  无向图存两遍
    h[b].push_back({a,c});  //与b相连的a的价值是c

  }
  //第一遍找到x的最远点
  dfs(1,-1,0); //从某一点开始（一号点），记录是从哪一个点过来的（-1），目前距离是（0）
  
  int u=1;
  for(int i=1;i<=n;i++)
    if(dist[i]>dist[u]) u=i;

//第二遍从最远点出发，找树的直径
  dfs(u,-1,0);

  for(int i=1;i<=n;i++)
    if(dist[i]>dist[u]) u=i;
 int v = dist[u];

//最后利用数学公式，输出花费
printf("%lld\n",10*v + v*(v+1ll)/2);
  return 0;
}