剑指offer-68.二叉树/二叉搜索树的最近公共祖先-CSDN博客

本文链接：https://blog.csdn.net/sinat_41473555/article/details/104922003

题目来源：

二叉搜索树的最近公共祖先 https://leetcode-cn.com/problems/er-cha-sou-suo-shu-de-zui-jin-gong-gong-zu-xian-lcof/

二叉树的最近公共祖先 https://leetcode-cn.com/problems/lowest-common-ancestor-of-a-binary-tree/

给定一个二叉搜索树, 找到该树中两个指定节点的最近公共祖先。

百度百科中最近公共祖先的定义为：“对于有根树 T 的两个结点 p、q，最近公共祖先表示为一个结点 x，满足 x 是 p、q 的祖先且 x 的深度尽可能大（一个节点也可以是它自己的祖先）。”

二叉树的最近公共祖先：

先看一般情况的二叉树，根据最近公共祖先的定义，有以下推论：

如果某节点node == p或node == q，那么node就是p和q的最近公共祖先。
如果节点node不满足第一条，那么从它的左右子树中继续寻找，直到找到一个节点，在它的左右子树中能找到p和q，那么该节点为最近公共祖先。

# Definition for a binary tree node.
# class TreeNode(object):
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None

class Solution(object):
    def lowestCommonAncestor(self, root, p, q):
        """
        :type root: TreeNode
        :type p: TreeNode
        :type q: TreeNode
        :rtype: TreeNode
        """
        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 left and right:
            return root
        elif left and not right:
            return left
        elif right and not left:
            return right

时间复杂度： O(n) ，最坏情况下需要遍历整棵树

空间复杂度： O(n) ，递归堆栈使用的空间可能达到n。

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

再看二叉搜索树的情况，其实可以用相同的题解，但是二叉搜索树有很好的性质，根的左节点都小于根，根的右节点都大于根。所以可以进行简化：

如果p.val和q.val都大于node.val，那么只需在node.right中继续寻找
如果p.val和q.val都小于node.val，那么只需在node.left中继续寻找
否则，说明p和q分布在node的左右两侧，那么node肯定是它们的最小公共祖先。

class Solution(object):
    def lowestCommonAncestor(self, root, p, q):
        """
        :type root: TreeNode
        :type p: TreeNode
        :type q: TreeNode
        :rtype: TreeNode
        """
        p_val = p.val
        q_val = q.val
        while root:
            if p_val > root.val and q_val > root.val:
                root = root.right
            if p_val < root.val and q_val < root.val:
                root = root.left
            else:
                return root

时间复杂度： O(n) ，最坏情况下需要遍历整棵树

空间复杂度： O(1) ，没有递归栈，所以只有常数个变量