蓝桥杯 历届试题 大臣的旅费（树的直径求解）

问题描述
很久以前，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最多花费的路费是多少。

样例输入1
5
1 2 2
1 3 1
2 4 5
2 5 4

样例输出1
135

解题思路
根据题目中这句话“同时，如果不重复经过大城市，从首都到达每个大城市的方案都是唯一的。”可知，城市之间不构成回路。即，城市所构成的是树状结构。题目需要求出所有城市中相隔最远的两个城市的距离，等价于求树的直径。
树的直径求解思路
1.两次DFS
先从任意节点出发，找到距离它最远的点x。
再从点x出发找到距离它最远的点y，点x和点y间的距离就是树的直径。

相关证明可以参考下面这篇博客
https://blog.csdn.net/forever_dreams/article/details/81051578

2.树形DP
从根节点向下，依次求出每个节点的到叶节点的最长距离和次长距离，注意最长距离和次长距离必须是经过该节点的两个不同子节点。那么树的直径就等于所有节点的最长距离和次长距离的和的最大值。

这篇博客中讲解的比较详细，且易懂。
https://blog.csdn.net/niiick/article/details/80708655

代码（两次dfs）

#include <stdio.h>
#include <vector>
using namespace std;
struct E{
    int next;
    int c;
};
vector<E> edge[100005];
bool mark[100005];
int root=0;
int maxlen=0;

void dfs(int start,int length)
{
    if(mark[start])return;
    mark[start]=true;
    for(int i=0;i<edge[start].size();i++)
    {
        //if(mark[edge[start][i].next])continue;
        int next=edge[start][i].next;
        int len=edge[start][i].c;
        //printf("next=%d,len=%d\n",next,len);
        dfs(next,length+len);
    }
    mark[start]=false;
    if(length>maxlen){
        maxlen=length;
        root=start;
        //printf("maxlen=%d,root=%d\n",maxlen,root);
    }
}
int main()
{
    int n;
    scanf("%d",&n);
    n--;
    while(n>0)
    {
        int a,b,c;
        scanf("%d%d%d",&a,&b,&c);
        E tmp;
        tmp.c=c;
        tmp.next=b;
        edge[a].push_back(tmp);
        tmp.next=a;
        edge[b].push_back(tmp);
        n--;
    }
    for(int i=0;i<n;i++){
        mark[i]=false;
    }
    dfs(1,0);
    //printf("%d\n",root);
    dfs(root,0);
    printf("%d\n",maxlen*10+(maxlen*(maxlen+1))/2);
}

运行结果