问题描述
很久以前,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);
}
运行结果