代码随想录算法训练营第二十二日｜LC235. 二叉搜索树的最近公共祖先 LC701.二叉搜索树中的插入操作 450.删除二叉搜索树中的节点

LC235二叉搜索树的最近公共祖先：

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.left, p, q)
        if root.val < p.val and root.val < q.val:
            return self.lowestCommonAncestor(root.right, p, q)
        return root

如果当前root比q和p都大，那证明公共节点在root的左子树里，相反，如果root比p和q小，那说明公共节点在root的右子树里。

LC701.二叉搜索树中的插入操作:

class Solution:
    def insertIntoBST(self, root: Optional[TreeNode], val: int) -> Optional[TreeNode]:
        if not root:
            return TreeNode(val)
        if root.val > val:
            # 将val插入至当前root的左子树中合适的位置
            # 并更新当前root的左子树为包含目标val的新左子树
            root.left = self.insertIntoBST(root.left, val)
        if root.val < val:
            # 将val插入至当前root的右子树中合适的位置
            # 并更新当前root的右子树为包含目标val的新右子树
            root.right = self.insertIntoBST(root.right, val)
        return root

450.删除二叉搜索树中的节点:

class Solution:
    def deleteNode(self, root: Optional[TreeNode], key: int) -> Optional[TreeNode]:
        if not root:
            return root
        if root.val < key:
            root.right = self.deleteNode(root.right, key)
        elif root.val > key:
            root.left = self.deleteNode(root.left, key)
        else:
            if not root.left:
                return root.right # 当前节点的左子树为空，返回当前的右子树
            if not root.right:
                return root.left  # 当前节点的右子树为空，返回当前的左子树
            # 左右子树都不为空，找到右孩子的最左节点 记为node
            node = root.right
            while node.left:
                node = node.left
            # 将当前节点的左子树挂在node的左孩子上
            node.left = root.left
            # 当前节点的右子树替换掉当前节点，完成当前节点的删除
            root = root.right
        return root