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

论坛 期权论坛 编程之家     
选择匿名的用户   2021-6-2 20:54   2058   0

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

百度百科中最近公共祖先的定义为:“对于有根树 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

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/

分享到 :
0 人收藏
您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

积分:3875789
帖子:775174
精华:0
期权论坛 期权论坛
发布
内容

下载期权论坛手机APP