python coding with ChatGPT 打卡第21天| 二叉树：最近公共祖先

Luna_M

已于 2024-02-20 17:45:38 修改

阅读量1.3k

点赞数 31

分类专栏： Python Coding with ChatGPT 文章标签： python chatgpt 开发语言 leetcode 算法数据结构

于 2024-02-20 15:48:33 首次发布

本文链接：https://blog.csdn.net/baidu_33000721/article/details/136185147

版权

Python Coding with ChatGPT 专栏收录该内容

25 篇文章 0 订阅

订阅专栏

二叉树的最近公共祖先

Key Points

后序遍历（左右中）就是天然的回溯过程，可以根据左右子树的返回值，来处理中节点的逻辑。
判断逻辑是如果递归遍历遇到q，就将q返回，遇到p 就将p返回，那么如果左右子树的返回值都不为空，说明此时的中节点，一定是q 和p 的最近祖先。

视频讲解

自底向上查找，有点难度

重点分析

方法一：递归法

递归的终止条件：

遇到空的话，因为树都是空了，所以返回空。
如果 root == q，或者 root == p，说明找到 q p ，则将其返回，这个返回值，后面在中节点的处理过程中会用到

单层处理逻辑：

如果left 和 right都不为空，说明此时root就是最近公共节点。这个比较好理解
如果left为空，right不为空，就返回right，说明目标节点是通过right返回的，反之依然。

def lowestCommonAncestor(root, p, q):
    if not root:
        return root
    if root == p or root == q:
        return root
    left = lowestCommonAncestor(root.left, p, q)
    right = lowestCommonAncestor(root.right, p, q)
    if left and right:
        return root
    if left:
        return left
    if right:
        return right
    return None

在这里插入图片描述

评价：
直接使用一次遍历来找到最近公共祖先会更高效。这种方法不需要存储所有父节点的路径，而是利用递归在回溯过程中直接找到LCA。

方法二：递归法
首先为每个节点找到其所有的父节点，然后比较这两个列表来找到最近的公共父节点

def find_all_father(root, p):
    res = []
    def find_father(root, p):
        if not root:
            return False
        if root == p or find_father(root.left, p) or find_father(root.right, p):
            res.append(root)
            return True
        return False
    find_father(root, p)
    return res

def lowestCommonAncestor(root, p, q):
    res1 = find_all_father(root, p)
    res2 = find_all_father(root, q)
    for i in res1:
        for j in res2:
            if i == j:
                return i

评价：
对于每个节点p和q，你的方法都会遍历整棵树来找到它们的所有父节点。这意味着树被遍历了两次。此外，对于找到的每对父节点，你使用了双层循环来比较它们，这会导致效率低下，尤其是在大树上。

方法三：迭代法
在这里插入图片描述

def lowestCommonAncestor(root, p, q):
    # 字典用于保存每个节点的父节点
    parent = {root: None}
    stack = [root]
    
    # 迭代遍历树，直到找到p和q
    while p not in parent or q not in parent:
        node = stack.pop()
        if node.left:
            parent[node.left] = node
            stack.append(node.left)
        if node.right:
            parent[node.right] = node
            stack.append(node.right)
    
    # 通过父节点字典构建p到根节点的路径
    ancestors = set()
    while p:
        ancestors.add(p)
        p = parent[p]
    
    # 检查q到根节点的路径中第一个出现在p路径集合中的节点
    while q not in ancestors:
        q = parent[q]
    return q

这个代码首先使用一个栈和一个字典来迭代遍历整棵树并记录每个节点的父节点。然后，它构建了从p到根节点的路径并保存在一个集合中。最后，它追溯q到根节点的路径，直到找到第一个也出现在p的路径集合中的节点。这个节点就是p和q的最近公共祖先。

评价：
这个方法避免了递归，但需要额外的存储空间来记录父节点信息，并且需要两次遍历：一次是构建父节点的映射，一次是构建从目标节点到根节点的路径。

二叉搜索树的最近公共祖先

Key Points

在二叉搜索树（BST）中找到两个指定节点的最近公共祖先（LCA）比在普通二叉树中要简单，因为你可以利用BST的性质：对于任何节点，左子树上的所有节点的值都小于该节点的值，右子树上的所有节点的值都大于该节点的值。这个性质允许我们使用迭代或递归的方式来快速定位LCA，而不需要像在普通二叉树中那样存储父节点或递归遍历整个树。

视频讲解

二叉搜索树找祖先

重点分析

方法一：递归

def lowestCommonAncestor(root, p, q):
    if p.val > root.val and q.val > root.val:
        return lowestCommonAncestor(root.right, p, q)
    if p.val < root.val and q.val < root.val:
        return lowestCommonAncestor(root.left, p, q)
    return root

在这里插入图片描述

方法二：迭代法

def lowestCommonAncestor(root, p, q):
    while root:
        if p.val < root.val and q.val < root.val:
            root = root.left
        elif p.val > root.val and q.val > root.val:
            root = root.right
        else:
            return root