235. 二叉搜索树的最近公共祖先
二叉搜索树是有序的,因为是有序树,所有 如果 中间节点是 q 和 p 的公共祖先,那么 中节点的数组 一定是在 [p, q]区间的。不用使用回溯,二叉搜索树自带方向性,可以方便的从上向下查找目标区间,遇到目标区间内的节点,直接返回。
class Solution:
def traversal(self, cur, p, q):
if cur == None:
return None
if cur.val>p.val and cur.val>q.val:
left=self.traversal(cur.left,p,q)
if left is not None:
return left
if cur.val<p.val and cur.val<q.val:
right=self.traversal(cur.right,p,q)
if right is not None:
return right
return cur
def lowestCommonAncestor(self, root, p, q):
"""
:type root: TreeNode
:type p: TreeNode
:type q: TreeNode
:rtype: TreeNode
"""
return self.traversal(root,p,q)
class Solution:
def lowestCommonAncestor(self, root, p, q):
if root.val > p.val and root.val > q.val:
return self.lowestCommonAncestor(root.left, p, q)
elif root.val < p.val and root.val < q.val:
return self.lowestCommonAncestor(root.right, p, q)
else:
return root
701.二叉搜索树中的插入操作
只要遍历二叉搜索树,找到空节点 插入元素就可以
class Solution: def insertIntoBST(self, root: Optional[TreeNode], val: int) -> Optional[TreeNode]: if not root: node=TreeNode(val) return node cur=root parent=root while cur: parent=cur if cur.val>val: cur=cur.left else: cur=cur.right node=TreeNode(val) if val<parent.val: parent.left=node elif val>parent.val: parent.right=node return root class Solution: def insertIntoBST(self, root: Optional[TreeNode], val: int) -> Optional[TreeNode]: if not root: node=TreeNode(val) return node if root.val>val: root.left=self.insertIntoBST(root.left,val) if root.val<val: root.right=self.insertIntoBST(root.right,val) return root
450.删除二叉搜索树中的节点
二叉搜索树中删除节点遇到的情况
- 第一种情况:没找到删除的节点,遍历到空节点直接返回了
- 找到删除的节点
- 第二种情况:左右孩子都为空(叶子节点),直接删除节点, 返回NULL为根节点
- 第三种情况:删除节点的左孩子为空,右孩子不为空,删除节点,右孩子补位,返回右孩子为根节点
- 第四种情况:删除节点的右孩子为空,左孩子不为空,删除节点,左孩子补位,返回左孩子为根节点
- 第五种情况:左右孩子节点都不为空,则将删除节点的左子树头结点(左孩子)放到删除节点的右子树的最左面节点的左孩子上,返回删除节点右孩子为新的根节点。
class Solution: def deleteNode(self, root, key): if root is None: # 如果根节点为空,直接返回 return root if root.val == key: # 找到要删除的节点 if root.right is None: # 如果右子树为空,直接返回左子树作为新的根节点 return root.left cur = root.right while cur.left: # 找到右子树中的最左节点 cur = cur.left root.val, cur.val = cur.val, root.val # 将要删除的节点值与最左节点值交换 root.left = self.deleteNode(root.left, key) # 在左子树中递归删除目标节点 root.right = self.deleteNode(root.right, key) # 在右子树中递归删除目标节点 return root