LeetCode--235--二叉树的最近公共祖先

问题描述:

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

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

示例 1:

输入: root = [6,2,8,0,4,7,9,null,null,3,5], p = 2, q = 8
输出: 6
解释: 节点 2 和节点 8 的最近公共祖先是 6。

示例 2:

输入: root = [6,2,8,0,4,7,9,null,null,3,5], p = 2, q = 4
输出: 2
解释: 节点 2 和节点 4 的最近公共祖先是 2, 因为根据定义最近公共祖先节点可以为节点本身。

说明:

  • 所有节点的值都是唯一的。
  • p、q 为不同节点且均存在于给定的二叉搜索树中。

由于是二叉搜索树,找到 p.val <= root.val <= q.val return root

方法1:

 class Solution(object):
def lowestCommonAncestor(self, root, p, q):
"""
:type root: TreeNode
:type p: TreeNode
:type q: TreeNode
:rtype: TreeNode
"""
if root == None:
return None
min_ = min(p.val,q.val)
max_ = max(p.val,q.val)
if min_<= root.val <= max_:
return root
else:
l = self.lowestCommonAncestor(root.left,p,q)
r = self.lowestCommonAncestor(root.right,p,q)
if not l :
return r
if not r :
return l

改:

 class Solution(object):
def lowestCommonAncestor(self, root, p, q):
"""
:type root: TreeNode
:type p: TreeNode
:type q: TreeNode
:rtype: TreeNode
"""
while root:
if p.val < root.val > q.val:
root = root.left
elif p.val > root.val < q.val:
root = root.right
else:
return root

方法2:(官方)

 class Solution(object):
def search(self, root, x, stack):
stack.append(root)
if x.val < root.val:
self.search(root.left, x, stack) # 一定存在
elif root.val < x.val:
self.search(root.right, x, stack)
else:
return
def lowestCommonAncestor(self, root, p, q):
"""
:type root: TreeNode
:type p: TreeNode
:type q: TreeNode
:rtype: TreeNode
"""
s_p, s_q = [], []
self.search(root, p, s_p)
self.search(root, q, s_q)
n = min(len(s_q), len(s_p))
for i in range(n):
if s_p[i] != s_q[i]:
return s_q[i - 1]
return s_q[n-1]

2018-09-21 06:56:21

上一篇:《你必须知道的.NET》读书笔记一:小OO有大智慧


下一篇:深入理解Java虚拟机之读书笔记一 自动内存管理机制