Leetcode 235 二叉搜索树的最近公共祖先
题目:
给定一个二叉搜索树, 找到该树中两个指定节点的最近公共祖先。
百度百科中最近公共祖先的定义为:“对于有根树 T 的两个结点 p、q,最近公共祖先表示为一个结点 x,满足 x 是 p、q 的祖先且 x 的深度尽可能大(一个节点也可以是它自己的祖先)。”
例如,给定如下二叉搜索树: root = [6,2,8,0,4,7,9,null,null,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 为不同节点且均存在于给定的二叉搜索树中。
算法思想:
- 递归法:
递归参数:二叉搜索树根节点
递归结束条件:如果当前root为None,返回None;如果当前root恰好为p(q)则返回p(q)[这里可能存在疑问,这里为什么就返回了?仔细读题,题目说明中说的很清楚,p,q均在二叉搜索树中。];最后一种情况就是,如果当前p(q)和q(p)分居root的左右两侧,则返回root,说明当前的root就是最近公共祖先。
一次递归逻辑:递归遍历左子树,递归遍历右子树,别忘了层层返回。 - 迭代法:
逻辑与递归法基本相似,二叉搜索树通常利用优秀的性质,采用中序遍历。但仔细想,本题是找到从下向上最近的公共节点,因此采用后序遍历。【为什么?后序遍历的顺序是左右中,因此可以模拟从下向上找的过程。】
代码实现:
- 递归法:
# Definition for a binary tree node.
# class TreeNode:
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
class Solution:
def traversal(self, root, p, q):
if root == p:
return p
if root == q:
return q
if root.val > p.val and root.val > q.val:
left = self.traversal(root.left, p, q)
if left:
return left
if root.val < p.val and root.val < q.val:
right = self.traversal(root.right, p, q)
if right:
return right
return root
def lowestCommonAncestor(self, root, p, q):
return self.traversal(root, p, q)
- 迭代法:
# Definition for a binary tree node.
# class TreeNode:
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
class Solution:
def lowestCommonAncestor(self, root, p, q):
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 None
Leetcode 701 二叉搜索树中的插入操作
题目:
给定二叉搜索树(BST)的根节点 root 和要插入树中的值 value ,将值插入二叉搜索树。 返回插入后二叉搜索树的根节点。 输入数据 保证 ,新值和原始二叉搜索树中的任意节点值都不同。
注意,可能存在多种有效的插入方式,只要树在插入后仍保持为二叉搜索树即可。 你可以返回 任意有效的结果 。
示例 1:
输入:root = [4,2,7,1,3], val = 5
输出:[4,2,7,1,3,5]
解释:另一个满足题目要求可以通过的树是:
示例 2:
输入:root = [40,20,60,10,30,50,70], val = 25
输出:[40,20,60,10,30,50,70,null,null,25]
示例 3:
输入:root = [4,2,7,1,3,null,null,null,null,null,null], val = 5
输出:[4,2,7,1,3,5]
提示:
- 树中的节点数将在 [0, 1 0 4 10^4 104]的范围内。
- - 1 0 8 10^8 108 <= Node.val <= 1 0 8 10^8 108
- 所有值 Node.val 是 独一无二 的。
- - 1 0 8 10^8 108 <= val <= 1 0 8 10^8 108
- 保证 val 在原始BST中不存在。
算法思想:
- 递归法:
按照二叉搜索树的条件,插入即可。
要明确一点,二叉搜索树插入节点有多种方式
但是向叶子节点插入数据是最简单的方式
只要插入一个节点,就会在二叉搜索树的叶子节点位置找到插入的位置 - 迭代法:
也是二叉搜索树的性质,也很简单
代码实现:
- 递归法:
# Definition for a binary tree node.
# class TreeNode:
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
class Solution:
def traversal(self, root, p, q):
if root == p:
return p
if root == q:
return q
if root.val > p.val and root.val > q.val:
left = self.traversal(root.left, p, q)
if left:
return left
if root.val < p.val and root.val < q.val:
right = self.traversal(root.right, p, q)
if right:
return right
return root
def lowestCommonAncestor(self, root, p, q):
return self.traversal(root, p, q)
- 迭代法:
# Definition for a binary tree node.
# class TreeNode:
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
class Solution:
def lowestCommonAncestor(self, root, p, q):
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 None