大臣的旅费（dfs）

[蓝桥杯 2013 省 A] 大臣的旅费

题目描述

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

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

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

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

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

输入格式

输入的第一行包含一个整数 $n(n\le 10^5)$，表示包括首都在内的 $T$ 王国的城市数。

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

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

每行三个整数 $P_i,Q,D_i$，表示城市 $P_i$ 和城市 $Q_i$ 之间有一条高速路，长度为 $D_i(D_i\le 1000)$ 米。

输出格式

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

样例 #1

样例输入 #1

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

样例输出 #1

135

提示

样例解释：大臣 J 从城市 $4$ 到城市 $5$ 要花费 $135$ 的路费。

时限 5 秒, 64M。蓝桥杯 2013 年第四届省赛

思路

树的直径一般用树形 DP 或两次 DFS 的方式求解，我这里使用的是两次 DFS。

第一次 DFS 从根向下，寻找离根最远的结点（可以证明一定是叶子结点）

第二次 DFS 从第一次找到的最远的节点开始，寻找离这个点最远的点，两者的距离就是树的直径。

代码

#include<iostream>
#include<algorithm>
#include<cstring>
using namespace std;

const int N=1e5+10;
struct edge{int v,w;};//v为这条边的终点，w为权重（距离）
vector<edge> e[N];
int dist[N];//存放当前节点最大值
int n;

void dfs(int u,int fa,int distance){
	dist[u]=distance;
	for(auto ed:e[u]){//遍历节点
		if(ed.v==fa) continue;//跳过父节点避免返回
		dfs(ed.v,u,ed.w+distance);
	}
}

int main() {
	cin>>n;
	for(int i=0;i<n-1;i++){
		int a,b,c;
		cin>>a>>b>>c;
		e[a].push_back({b,c});
		e[b].push_back({a,c});
	}
	dfs(1,-1,0);//可任意找一点开始dfs，找到该点能到达的所有点
	
	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 s=dist[u];
	printf("%lld",10*s+s*(s+1ll)/2);//放溢出
	
	return 0;
}