树的直径求解方法

目录

  • [1 介绍](#1 介绍)
  • [2 训练](#2 训练)

1 介绍

树的直径是指树中任意两个节点之间的最长路径长度。通常可以通过两次广度优先搜索(BFS)或深度优先搜索(DFS)来求树的直径:

  1. 第一次从任意一点出发,找到离它最远的节点 A。
  2. 第二次从节点 A 出发,找到离它最远的节点 B,A 和 B 之间的路径长度就是树的直径。

2 训练

题目1100318. 合并两棵树后的最小直径

解题思路:分别求出两棵树的直径,然后比较大小max(d1, d2, r1 + r2 + 1)即可。

python3代码如下,

py 复制代码
from collections import deque

class Solution:
    def minimumDiameterAfterMerge(self, edges1: List[List[int]], edges2: List[List[int]]) -> int:

        def bfs(graph, start):
            n = len(graph)
            dist = [-1] * n
            queue = deque([start])
            dist[start] = 0
            while queue:
                node = queue.popleft()
                for neighbor in graph[node]:
                    if dist[neighbor] == -1:
                        dist[neighbor] = dist[node] + 1
                        queue.append(neighbor)
            farthest_node = dist.index(max(dist))
            return farthest_node, dist

        def find_diameter_and_radius(graph):
            node1, _ = bfs(graph, 0)
            node2, dist = bfs(graph, node1)
            diameter = dist[node2]
            return diameter, dist

        n = len(edges1) + 1 
        m = len(edges2) + 1
        graph1 = [[] for _ in range(n)]
        graph2 = [[] for _ in range(m)]
        
        for u, v in edges1:
            graph1[u].append(v)
            graph1[v].append(u)
            
        for u, v in edges2:
            graph2[u].append(v)
            graph2[v].append(u)
            
        diameter1, dist1 = find_diameter_and_radius(graph1)
        diameter2, dist2 = find_diameter_and_radius(graph2)
        
        # Finding radius of each tree
        radius1 = (diameter1 + 1) // 2
        radius2 = (diameter2 + 1) // 2
        
        # Minimum new diameter
        min_diameter = max(diameter1, diameter2, radius1 + radius2 + 1)
        
        return min_diameter
相关推荐
CHANG_THE_WORLD12 分钟前
if条件语句 三目运算符 汇编分析
汇编·算法·条件语句·if 语句·汇编分析·条件语句汇编分析
tumu_C13 分钟前
无用知识研究:在trailing return type利用decltype,comma operator在对函数进行sfinae原创 [二]
开发语言·c++·算法
web安全工具库19 分钟前
告别刀耕火种:用 Makefile 自动化 C 语言项目编译
linux·运维·c语言·开发语言·数据库·算法·自动化
海蓝可知天湛38 分钟前
利用Genspark自定义智能体:算法竞赛测试数据反推与生成工具
算法·aigc
BothSavage42 分钟前
Qwen3-VL-8B-Instruct推理测试transformer+sglang双版本
算法
尤超宇1 小时前
YOLOv3 目标检测算法核心技术
算法·yolo·目标检测
cyclel1 小时前
散列表的小想法
算法
Code小翊1 小时前
堆的基础操作,C语言示例
java·数据结构·算法
余俊晖1 小时前
如何让多模态大模型学会“自动思考”-R-4B训练框架核心设计与训练方法
人工智能·算法·机器学习
Emilia486.2 小时前
【Leetcode&nowcode&数据结构】顺序表的应用
数据结构·算法·leetcode