235. 二叉搜索树的最近公共祖先
每日几道leetcode刷刷题!
传送门
题目描述
给定一个二叉搜索树, 找到该树中两个指定节点的最近公共祖先。
百度百科中最近公共祖先的定义为:“对于有根树 T 的两个结点 p、q,最近公共祖先表示为一个结点 x,满足 x 是 p、q 的祖先且 x 的深度尽可能大(一个节点也可以是它自己的祖先)。”
示例 1:
代码
递归法
class Solution:
def lowestCommonAncestor(self, root: 'TreeNode', p: 'TreeNode', q: 'TreeNode') -> 'TreeNode':
def traveback(cur, p, q):
if not cur:
return cur
if cur.val > p.val and cur.val > q.val:
left = traveback(cur.left, p, q)
if left:
return left
if cur.val < p.val and cur.val < q.val:
right = traveback(cur.right, p, q)
if right:
return right
return cur # 在中间,说明cur就是最近公共祖先
return traveback(root, p, q)
迭代法
class Solution:
def lowestCommonAncestor(self, root: 'TreeNode', p: 'TreeNode', q: 'TreeNode') -> 'TreeNode':
while root:
if root.val > p.val and root.val > q.val:
root = root.left
elif root.val < p.val and root.val < q.val:
root = root.right
else:
return root
return