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

最新推荐文章于 2024-10-06 14:10:45 发布

996冲冲冲

最新推荐文章于 2024-10-06 14:10:45 发布

阅读量91

点赞数

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

本文链接：https://blog.csdn.net/weixin_44392348/article/details/125261453

版权

二叉搜索树专栏收录该内容

17 篇文章 0 订阅

订阅专栏

二叉搜索树性质决定了：如果 p.val 和 q.val 都比 root.val 小，则p、q肯定在 root 的左子树。
那问题规模就变小了，递归左子树就行！
如果 p.val 和 q.val 都比 root.val 大，递归右子树就行！
其他情况，root 即为所求：
只要 p.val 和 q.val 不是都大于(小于) root.val，即只要 p, q 不同处在 root 的一个子树
就只有这三种情况：
p 和 q 分居 root 的左、右子树。
root 就是 p，q 在 p 的子树中。
root 就是 q，p 在 q 的子树中
而这三种情况，p 和 q 的最近公共祖先都是 root

def lowestCommonAncestor(self, root, p, q):
        """
        :type root: TreeNode
        :type p: TreeNode
        :type q: TreeNode
        :rtype: TreeNode
        """
        def recur(root,p,q):                           #找到就返回
            if p.val < root.val and q.val < root.val:
                left = recur(root.left,p,q)
                if left:
                    return left
            elif p.val > root.val and q.val > root.val:
                right = recur(root.right,p,q)
                if right:
                    return right
            return root
        return recur(root, p, q)

一样的

def lowestCommonAncestor(self, root, p, q):
        """
        :type root: TreeNode
        :type p: TreeNode
        :type q: TreeNode
        :rtype: TreeNode
        """
        def recur(root,p,q):
            if p.val > root.val and q.val > root.val:
                return recur(root.right,p,q)
            elif p.val < root.val and q.val < root.val:
                return recur(root.left,p,q)
            return root
        return recur(root, p, q)

迭代写法

def lowestCommonAncestor(self, root, p, q):
        """
        :type root: TreeNode
        :type p: TreeNode
        :type q: TreeNode
        :rtype: TreeNode
        """
        while root:
            if root.val > q.val and root.val > p.val:
                root = root.left
            elif root.val < q.val and root.val < p.val:
                root = root.right
            else:
                break
        return root