【剑指offer】JZ68 二叉搜索树的最近公共祖先

1 问题

给定一个二叉搜索树, 找到该树中两个指定节点的最近公共祖先。
1.对于该题的最近的公共祖先定义:对于有根树T的两个节点p、q，最近公共祖先LCA(T,p,q)表示一个节点x，满足x是p和q的祖先且x的深度尽可能大。在这里，一个节点也可以是它自己的祖先.
2.二叉搜索树是若它的左子树不空，则左子树上所有节点的值均小于它的根节点的值； 若它的右子树不空，则右子树上所有节点的值均大于它的根节点的值
3.所有节点的值都是唯一的。
4.p、q 为不同节点且均存在于给定的二叉搜索树中。
数据范围:
3<=节点总数<=10000
0<=节点值<=10000

如果给定以下搜索二叉树: {7,1,12,0,4,11,14,#,#,3,5}，如下图:

示例1
输入：{7,1,12,0,4,11,14,#,#,3,5},1,12
返回值：7
说明：节点1 和 节点12的最近公共祖先是7

示例2
输入：{7,1,12,0,4,11,14,#,#,3,5},12,11
返回值：12
说明：因为一个节点也可以是它自己的祖先.所以输出12

2 答案

这题直接不会

官方解

1. 搜索路径比较

二叉搜索树没有相同值的节点，因此分别从根节点往下利用二叉搜索树较大的数在右子树，较小的数在左子树，可以轻松找到p、q。
直接得到从根节点到两个目标节点的路径，这样我们利用路径比较就可以找到最近公共祖先。

• step 1：根据二叉搜索树的性质，从根节点开始查找目标节点，当前节点比目标小则进入右子树，当前节点比目标大则进入左子树，直到找到目标节点。这个过程成用数组记录遇到的元素。
• step 2：分别在搜索二叉树中找到p和q两个点，并记录各自的路径为数组。
• step 3：同时遍历两个数组，比较元素值，最后一个相等的元素就是最近的公共祖先。

class Solution:
    def getPath(self, root, target):
        path = []
        node = root
        while node.val != target:
            path.append(node.val)
            if target < node.val:
                node = node.left
            else:
                node = node.right
        path.append(node.val)
        return path

    def lowestCommonAncestor(self , root: TreeNode, p: int, q: int) -> int:
        path_p = self.getPath(root, p)
        path_q = self.getPath(root, q)
        i = 0
        while i < len(path_p) and i < len(path_q):
            if path_p[i] == path_q[i]:
                res = path_p[i]
                i+=1
            else:
                break
        return res

2. 一次遍历

我们也可以利用二叉搜索树的性质：对于某一个节点若是p与q都小于等于这个这个节点值，说明p、q都在这个节点的左子树，而最近的公共祖先也一定在这个节点的左子树；若是p与q都大于等于这个节点，说明p、q都在这个节点的右子树，而最近的公共祖先也一定在这个节点的右子树。而若是对于某个节点，p与q的值一个大于等于节点值，一个小于等于节点值，说明它们分布在该节点的两边，而这个节点就是最近的公共祖先，因此从上到下的其他祖先都将这个两个节点放到同一子树，只有最近公共祖先会将它们放入不同的子树，每次进入一个子树又回到刚刚的问题，因此可以使用递归。

• step 1：首先检查空节点，空树没有公共祖先。
• step 2：对于某个节点，比较与p、q的大小，若p、q在该节点两边说明这就是最近公共祖先。
• step 3：如果p、q都在该节点的左边，则递归进入左子树。
• step 4：如果p、q都在该节点的右边，则递归进入右子树。

class Solution:
    def lowestCommonAncestor(self , root: TreeNode, p: int, q: int) -> int:
        if not root:
            return -1
        if p >= root.val and q <= root.val or q >= root.val and p <= root.val:  # 要有等号
            return root.val
        elif p > root.val and q > root.val:
            return self.lowestCommonAncestor(root.right, p, q)
        elif p < root.val and q < root.val:
            return self.lowestCommonAncestor(root.left, p, q)

https://www.nowcoder.com/share/jump/9318638301700804291989