Python实现"二叉搜索树的最近公共祖先"的一种方法

最新推荐文章于 2024-07-05 22:37:42 发布

求兵

最新推荐文章于 2024-07-05 22:37:42 发布

阅读量783

点赞数

分类专栏： Algorithms

本文链接：https://blog.csdn.net/qiubingcsdn/article/details/82897901

版权

Algorithms 专栏收录该内容

106 篇文章 5 订阅

订阅专栏

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

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

例如，给定如下二叉搜索树: root = [6,2,8,0,4,7,9,null,null,3,5]

     _______6______
       /              \
    ___2__          ___8__
   /      \        /      \
   0      _4       7       9
         /  \
         3   5

Example 1:

Input: root = [6,2,8,0,4,7,9,null,null,3,5], p = 2, q = 8
Output: 6
Explanation: The LCA of nodes 2 and 8 is 6.

Example 2:

Input: root = [6,2,8,0,4,7,9,null,null,3,5], p = 2, q = 4
Output: 2
Explanation: The LCA of nodes 2 and 4 is 2,since a node can be a descendant of itself 
             according to the LCA definition.

注意：

数中所有的结点值都是唯一的

q和p是不同的，并且在数中均存在

1：基本规则

当前结点等于q或者p，那么该结点必为公共结点
由于二叉搜索数左小右大的特定，判断当前结点的值与p和q的大小关系，假设p的值比q大。如果当前结点值满足q<value<p，当前结点必为公共结点
value>p，二叉树向下遍历左子树
value<q，二叉树向下遍历右子树

def lowestCommonAncestor(self, root, p, q):
        """
        :type root: TreeNode
        :type p: TreeNode
        :type q: TreeNode
        :rtype: TreeNode
        """
        if p.val < q.val:          #p大，q小
            p, q = q, p
        while root:
            # if root == p or root == q:
            #     return root
            # if root.val < p.val and root.val > q.val:
            #     return root
            if root.val < q.val:
                root = root.right
            if root.val > p.val:
                root = root.left
            else:      #该句else等价于上面被注释掉的四句
                return root

算法题来自：https://leetcode-cn.com/problems/lowest-common-ancestor-of-a-binary-search-tree/description/