树的直径(节点的最大距离)

树的直径是指相隔距离最大的两个节点间的距离,这两个节点一般是叶子节点,也有可能一个端点是根节点。
求树的直径的做法:可以选取任意一点遍历整棵树,找到最远点,该点一定是直径的一个端点,再从该点再次遍历整棵树,找到最远点,则该点一定是直径的另一个端点。
这里可以看该博主的算法证明:链接
例题:
链接
问题描述
很久以前,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;
}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
要计算中两个节点之间的距离,可以使用深度优先搜索(DFS)或广度优先搜索(BFS)算法。以DFS为例,我们可以从根节点开始,依次遍历每个节点,并记录每个节点的深度(即距离节点距离)。当我们遍历到目标节点时,就可以根据记录的深度计算出两个节点之间的距离。 下面是一个示例代码,用于计算中两个节点之间的距离: ```python # 定义一个函数,用于计算中两个节点之间的距离 def get_distance(root, node1, node2): # 记录每个节点的深度 depths = {} # 记录每个节点的父节点 parents = {} # 从根节点开始进行深度优先搜索 def dfs(node, depth, parent): depths[node] = depth parents[node] = parent # 如果找到了目标节点,就返回 if node == node1 or node == node2: return # 否则继续遍历子节点 for child in node.children: dfs(child, depth + 1, node) dfs(root, 0, None) # 计算两个节点距离 distance = 0 while node1 != node2: if depths[node1] < depths[node2]: node1, node2 = node2, node1 node1 = parents[node1] distance += 1 return distance ``` 在这个函数中,我们首先遍历,记录每个节点的深度和父节点,然后通过一个while循环,计算两个节点之间的距离。在循环中,我们通过比较两个节点的深度,不断向上移动深度更深的节点,直到两个节点的深度相等为止。在每次移动节点时,我们都会将距离加1。最后,函数返回的就是两个节点之间的距离。 需要注意的是,这个函数假设中不存在环,即任意两个节点之间只有一条路径。如果存在环,那么计算两个节点之间的距离就比较复杂,可能需要使用更复杂的算法。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值