【二叉树】具有所有最深节点的最小子树

豪冷啊

于 2022-06-14 15:12:55 发布

阅读量225

点赞数 1

分类专栏：算法文章标签： swift 二叉树算法深度优先

本文链接：https://blog.csdn.net/xjh093/article/details/125278400

版权

算法专栏收录该内容

121 篇文章 0 订阅

订阅专栏

这篇博客介绍了如何解决寻找给定二叉树中最深节点的子树问题。通过从根节点开始，计算左右子树的最大深度，当左右深度相等时，当前节点即为最深节点。如果深度不等，则进入深度较大的子树继续查找。提供的Swift解法有效地实现了这一过程。

摘要由CSDN通过智能技术生成

0x00 题目

给定一个根为 root 的二叉树
每个节点的深度是该节点到根的最短距离
返回包含原始树中所有 最深节点 的 最小子树
如果一个节点在 整个树 的任意节点之间具有最大的深度
则该节点是 最深的
一个节点的 子树 是该节点加上它的所有后代的集合

另一个类似的题是：
最深叶节点的最近公共祖先

给你一个有根节点 root 的二叉树
返回它 最深的叶节点的 最近 公共祖先

叶节点是二叉树中没有子节点的节点
树的根节点的深度为 0
如果某一节点的深度为 d
它的子节点的深度就是 d+1
如果假定 A 是一组节点 S 的 最近公共祖先
S 中的每个节点都在以 A 为根节点的子树中
且 A 的深度达到此条件下可能的最大值

0x01 思路

从根节点开始
获得当前节点的左右子树的最大深度
深度相同，说明最深的节点在这个节点两边
那这个节点就是结果
如果深度不相同，则去深度大的子树继续判断
最终就能得到结果

0x02 解法

语言：Swift

树节点：TreeNode

public class TreeNode {
    public var val: Int
    public var left: TreeNode?
    public var right: TreeNode?
    public init() { self.val = 0; self.left = nil; self.right = nil; }
    public init(_ val: Int) { self.val = val; self.left = nil; self.right = nil; }
    public init(_ val: Int, _ left: TreeNode?, _ right: TreeNode?) {
        self.val = val
        self.left = left
        self.right = right
    }
}

解法：

func subtreeWithAllDeepest(_ root: TreeNode?) -> TreeNode? {
    guard let r = root else { return nil }
    
    // 求某个节点的最大深度
    func maxDep(_ root: TreeNode?) -> Int {
        guard let r = root else { return 0 }
        return max(maxDep(r.left), maxDep(r.right)) + 1
    }
    
    // 左子树深度
    let left = maxDep(r.left)
    // 右子树深度
    let right = maxDep(r.right)
    
    // 深度相同，则当前节点为 最深叶节点的最近公共祖先
    if left == right { return r }
    
    // 不相同时，去深度大的子树查找
    if left > right {
        return subtreeWithAllDeepest(r.left)
    }
    
    return subtreeWithAllDeepest(r.right)
}