很久以前,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≤n≤1e5,
1≤Pi,Qi≤n,
1≤Di≤1000
输入样例:
5
1 2 2
1 3 1
2 4 5
2 5 4
输出样例:
135
思路:
从题目出发:
如果不重复经过大城市,从首都到达每个大城市的方案都是唯一的。说明该图是一个没有环的图,即一棵树。
费用的算法:
设距离为 x ,那么总费用 s = ( 1 + 10 ) + ( 2 + 10 ) + …… + ( x + 10 ) 化简得: s = 10 * x + x * ( 1 + x ) ;
最大距离(树的直径)算法:
任找一点 x ,开始遍历,找到距其最长的路径(端点记为 u),再从 u 点找到距其最长的路径,所求路径就是最大距离 x , 而 u 则为树的直径的端点。
算法证明:
反证法:假设 y 不是树的直径的端点, u --- v 是树的直径。
前提条件:x---y 为算法求出的路径, 即y为直径的端点, 而u --- v 为树的直径
第一种情况: x---y 与 u---v有交点, 假设 y 不是树的直径的端点:
因为 x --- y 是算法所求出的最长路径,则 x---u 的距离 一定小于等于 x --- y 的距离
即 ① + ② ≤ ① + ④ ----> ② ≤ ④
而 u --- v 又是树的直径,
则有 ③ + ② ≥ ③ + ④ ----> ② ≥ ④
与假设矛盾
第二种情况: x --- y 与 u --- v 之间没有交点,假设 y 不是树的直径的端点:
因为 x --- y 为算法所求最大路径,
则 x --- y 的距离 大于等于 x --- v 的路径
① + ② ≥ ① + ③ + ⑤ ---> ② ≥ ③ + ⑤ ----> ② > ⑤
而 u --- v 又是树的直径
④ + ⑤ ≥ ④ + ③ + ② ---> ⑤ ≥ ③ + ② ---> ⑤ > ②
与假设矛盾;
综上,算法所求的端点y,一定是直径的端点。
完整代码(C++):
#include<algorithm>
#include<cstdio>
#include<cstring>
#include<iostream>
using namespace std;
const int N = 1e5 + 10;
struct Edge
{
int id;
int w;
};
int n;
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(node.id, u, distance + node.w);
}
}
}
int main()
{
cin >> 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});
h[b].push_back({a, c});
}
dfs(1, -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 s = dist[u];
cout << 10 * s + s * (1ll + s) / 2 << endl;
return 0;
}