剑指 Offer 68 - II. 二叉树的最近公共祖先

愈努力俞幸运

于 2022-07-31 22:13:28 发布

阅读量1.2k

点赞数

分类专栏： python习题练习文章标签： leetcode 算法职场和发展

本文链接：https://blog.csdn.net/qq_37891604/article/details/126091174

版权

python习题练习专栏收录该内容

61 篇文章 1 订阅

订阅专栏

该博客讨论了如何使用递归实现二叉树的最近公共祖先（LCA）问题。通过先序遍历，当遇到节点p或q时返回，并在回溯过程中判断p和q是否位于当前节点的异侧，从而找到最近公共祖先。递归终止条件包括到达叶子节点、找到p或q。返回值取决于左右子树的返回结果，根据四种情况判断最近公共祖先。

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

剑指 Offer 68 - II. 二叉树的最近公共祖先https://leetcode.cn/problems/er-cha-shu-de-zui-jin-gong-gong-zu-xian-lcof/

考虑通过递归对二叉树进行先序遍历，当遇到节点 p或 q 时返回。从底至顶回溯，当节点 p, q 在节点 root的异侧时，节点 root 即为最近公共祖先，则向上返回 root 。

想上回溯什么？

递归解析：
终止条件：
当越过叶节点，则直接返回 null ；
当 root等于 p,或q，则直接返回 root ；
递推工作：
开启递归左子节点，返回值记为 left ；
开启递归右子节点，返回值记为 right ；
返回值：根据 left 和 right，可展开为四种情况；
1.当 left 和 right同时为空：说明 root的左 / 右子树中都不包含 p,q，返回 null ；
2.当 left和 right 同时不为空：说明 p, q分列在 root的异侧（分别在左 / 右子树），因此 root 为最近公共祖先，返回 root ；
3. 当 left为空，right 不为空：p,q都不在 root的左子树中，直接返回 right。具体可分为两种情况：
（1）p,q其中一个在 root的右子树中，假设为p, 此时 right 指向 p；
(2)p,q两节点都在 rootroot 的右子树中，此时的 right指向最近公共祖先节点；
4.当 left不为空， right为空：与情况 3. 同理；

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