【Python实战】LeetCode235、236：验证二叉树的最近公共祖先

LeetCode236：二叉树的最近公共祖先

递归求解

# 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: # 如果树为空，返回none；同时也作为递归的终止条件之一
            return None
        if root == p or root == q: # 如果 p和q中有等于root的，那么它们的最近公共祖先即为root；同时也作为递归的终止条件之一
            return root
        left = self.lowestCommonAncestor(root.left, p, q)  # 递归遍历左子树，只要在左子树中找到了p或q，则先找到谁就返回谁
        right = self.lowestCommonAncestor(root.right, p, q) # 递归遍历右子树，只要在右子树中找到了p或q，则先找到谁就返回谁
        if not left: # 如果在左子树中，p和 q都找不到，则 p和 q一定都在右子树中，先遍历到的那个就是最近公共祖先，因为一个节点也可以是它自己的祖先
            return right
        elif not right: # 如果在左子树中有找到节点（p或q），这时候要再判断一下右子树中的情况，如果在右子树中，p和q都找不到，则 p和q一定都在左子树中
            return left
        else: # 如果在左子树和右子树中都有找到节点，则 p和 q一个存在于左子树，一个存在于右子树，最近公共祖先只能为root
            return root

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

上一题的代码适用于所有的二叉树，也可以套用到这题中，只不过，对于二叉搜索树来说，它有一些特殊的性质可以利用，即每个节点的左子树的所有节点都小于该节点，右子树的所有节点都大于该节点，因此，在判断 p 和 q 这两个不同的节点位于左子树还是右子树中时，可通过比较数值关系。

解法一：递归法

class Solution(object):
    def lowestCommonAncestor(self, root, p, q):
        """
        :type root: TreeNode
        :type p: TreeNode
        :type q: TreeNode
        :rtype: TreeNode
        """
        if p.val < root.val and q.val < root.val: # 如果 p和 q都位于左子树中，则去左子树中继续找
            return self.lowestCommonAncestor(root.left, p, q)
        if p.val > root.val and q.val > root.val: # 如果 p和 q都位于右子树中，则去右子树中继续找
            return self.lowestCommonAncestor(root.right, p, q)
        return root # 如果 p和 q一个位于左子树，一个位于右子树中，则返回根节点

解法二：迭代法

class Solution(object):
    def lowestCommonAncestor(self, root, p, q):
        """
        :type root: TreeNode
        :type p: TreeNode
        :type q: TreeNode
        :rtype: TreeNode
        """
        while root: # 注意 root的值也在不断更新的，所以是不断向下遍历的过程
            if p.val < root.val and q.val < root.val: # 如果 p和 q都位于左子树中，则去左子树中继续找
                root = root.left
            elif p.val > root.val and q.val > root.val: # 如果 p和 q都位于右子树中，则去右子树中继续找
                root = root.right
            else: # 如果 p和 q一个位于左子树，一个位于右子树中，则返回根节点
                return root

最后，如果大家有更好的Python解法，欢迎在评论区分享下您的解法，一起进步，感谢^ o ^~