树的直径是指相隔距离最大的两个节点间的距离,这两个节点一般是叶子节点,也有可能一个端点是根节点。
求树的直径的做法:可以选取任意一点遍历整棵树,找到最远点,该点一定是直径的一个端点,再从该点再次遍历整棵树,找到最远点,则该点一定是直径的另一个端点。
这里可以看该博主的算法证明:链接
例题:
链接
问题描述
很久以前,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
输出格式
大臣J从城市4到城市5要花费135的路费。
c++代码:
//这里是通过1号节点去找到一个直径端点,再通过该端点找到另一端点
#include<iostream>
#include<vector>
using namespace std;
#define max_size 10010
typedef struct
{
long len;
int node;
}leaf;
vector<int> adj[max_size];//用于装相邻节点
vector<int> dis[max_size];//用于装距离
vector<leaf> op;//用于记录从1号点或从1号节点的最远节点 出发到对应下标叶子节点的距离
leaf base;//用于装找到的最大距离的节点下标和距离
void create_tree(int n)
{
int i,a,b,c;
for(i=1;i<n;i++)
{
cin>>a>>b>>c;
adj[a].push_back(b);
dis[a].push_back(c);
adj[b].push_back(a);
dis[b].push_back(c);
}
}
void dfs(int num,long sum,int pre)
{
if(adj[num].size()==1&&adj[num][0]==pre)//相邻节点只有上一个节点,即到叶子节点
{
leaf t;
t.len=sum;
t.node=num;
op.push_back(t);
}
else
{
int i;
long a;
for(i=0;i<adj[num].size();i++)
{
a=sum+dis[num][i];
if(adj[num][i]==pre) continue;
dfs(adj[num][i],a,num);
}
}
}
int calculate()
{
int i;
base.len=op[0].len;
base.node=op[0].node;
for(i=1;i<op.size();i++)
{
if(op[i].len>base.len)
{
base.len=op[i].len;
base.node=op[i].node;
}
}
return base.node;
}
int main()
{
int n;
cin>>n;
create_tree(n);
dfs(1,0,0);
int node,num;
node=calculate();//找到距离1号节点最远的点,即先找到直径的一个端点
dfs(node,0,0);//通过已知端点,找到另一端点
num=calculate();
long res;
res=base.len;
res=res*(21+res)/2;//等差数列求和,计算结果
cout<<res;
}