给定一个二叉搜索树, 找到该树中两个指定节点的最近公共祖先。
百度百科中最近公共祖先的定义为:“对于有根树 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/