LeetCode 235. 二叉搜索树的最近公共祖先
力扣题目链接
利用二叉搜索树的性质,
- 如果根结点值小于p和q,那么最近公共祖先在右子树,
- 如果根结点值大于p和q,那么最近公共祖先在左子树,
- 否则,根结点就是最近公共祖先
class Solution:
def lowestCommonAncestor(self, root: 'TreeNode', p: 'TreeNode', q: 'TreeNode') -> 'TreeNode':
if root.val < p.val and root.val < q.val:
return self.lowestCommonAncestor(root.right, p, q)
elif root.val > p.val and root.val > q.val:
return self.lowestCommonAncestor(root.left, p, q)
else:
return root
LeetCode 701.二叉搜索树中的插入操作
力扣题目链接
在二叉搜索树中,一定可以找到一个叶子节点的位置插入元素
- 如果根结点为空,直接返回要插入的节点
- 如果当前节点值小于要插入的值,则往右子树中插,若同时右子树为空,则可直接插入到右子树上
- 大于同理
class Solution:
def insertIntoBST(self, root: Optional[TreeNode], val: int) -> Optional[TreeNode]:
if not root:
return TreeNode(val)
if root.val < val:
if root.right is None:
root.right = TreeNode(val)
else:
self.insertIntoBST(root.right, val)
else:
if root.left is None:
root.left = TreeNode(val)
else:
self.insertIntoBST(root.left, val)
return root
LeetCode 450.删除二叉搜索树中的节点
- 第一种情况:没找到删除的节点,遍历到空节点直接返回了
- 找到删除的节点
- 第二种情况:左右孩子都为空(叶子节点),直接删除节点, 返回NULL为根节点
- 第三种情况:删除节点的左孩子为空,右孩子不为空,删除节点,右孩子补位,返回右孩子为根节点
- 第四种情况:删除节点的右孩子为空,左孩子不为空,删除节点,左孩子补位,返回左孩子为根节点
- 第五种情况:左右孩子节点都不为空,则将删除节点的左子树头结点(左孩子)放到删除节点的右子树的最左面节点的左孩子上,返回删除节点右孩子为新的根节点。
class Solution:
def deleteNode(self, root: Optional[TreeNode], key: int) -> Optional[TreeNode]:
if not root: return root
if root.val == key:
if root.left is None:
return root.right
elif root.right is None:
return root.left
else:
cur = root.right
while cur.left:
cur = cur.left
cur.left = root.left
return root.right
if root.val > key:
root.left = self.deleteNode(root.left, key)
elif root.val < key:
root.right = self.deleteNode(root.right, key)
return root
今日毕!