LeetCode刷题——删除二叉搜索树中的节点#450#Medium

删除二叉搜索树中的节点的思路探讨与源码
    删除二叉搜索树中的节点的题目如下图，该题属于树类型和搜索类型的题目，主要考察对于搜索方法的使用和树结构的理解。本文的题目作者想到2种方法，分别是迭代方法和递归方法，其中迭代方法使用Java进行编写，而递归方法使用Python进行编写，当然这可能不是最优的解法，还希望各位大佬给出更快的算法。

    本人认为该题目可以使用迭代方法的思路进行解决，首先初始化参数，判断是否为空并开始遍历，遍历过程中判断树的节点的元素是否大于key值，如果是就继续搜索左子树，否则就搜索右子树，按这个规则直到遍历结束。判断结束后的树节点是否为空，如果是就返回根节点。判断树的左子树或者右子树是否同时为空，如果是就让节点等于空值，如果只是右子树为空就把节点指向左节点，如果只是左子树为空就把节点指向右节点，如果都不为空就继续进行搜索，按照这个思路直到迭代结束并返回结果。那么按照这个思路我们的Java代码如下：

#喷火龙与水箭龟
class Solution {
    public TreeNode deleteNode(TreeNode root, int key) {
        TreeNode cur = root;
        TreeNode curParent = null;
        while (cur != null && cur.val != key) {
            curParent = cur;
            if (cur.val > key) {
                cur = cur.left;
            } else {
                cur = cur.right;
            }
        }
        if (cur == null) {
            return root;
        }
        if (cur.left == null && cur.right == null) {
            cur = null;
        } else if (cur.right == null) {
            cur = cur.left;
        } else if (cur.left == null) {
            cur = cur.right;
        } else {
            TreeNode successor = cur.right, successorParent = cur;
            while (successor.left != null) {
                successorParent = successor;
                successor = successor.left;
            }
            if (successorParent.val == cur.val) {
                successorParent.right = successor.right;
            } else {
                successorParent.left = successor.right;
            }
            successor.right = cur.right;
            successor.left = cur.left;
            cur = successor;
        }
        if (curParent == null) {
            return cur;
        } else {
            if (curParent.left != null && curParent.left.val == key) {
                curParent.left = cur;
            } else {
                curParent.right = cur;
            }
            return root;
        }
    }
}

    显然，我们的迭代方法的效果不错，还可以用递归方法进行解决。首先判断树是否为空，如果是就返回空；再判断树的根节点的值是否大于key值，如果是就对左子树进行递归操作。如果树的根节点的值小于key值，就对右子树进行递归操作。如果左子树或者右子树有一个是空的，那么就把非空的子树赋值到当前节点。如果上述条件都不满足，则继续进行递归操作，直到递归满足终止条件后结束并返回结果。所以按照这个思路就可以解决，下面是Python代码：

#喷火龙与水箭龟
class Solution:
    def deleteNode(self, root: Optional[TreeNode], key: int) -> Optional[TreeNode]:
        if root is None:
            return None
        if root.val > key:
            root.left = self.deleteNode(root.left, key)
        elif root.val < key:
            root.right = self.deleteNode(root.right, key)
        elif root.left is None or root.right is None:
            root = root.left if root.left else root.right
        else:
            successor = root.right
            while successor.left:
                successor = successor.left
            successor.right = self.deleteNode(root.right, successor.val)
            successor.left = root.left
            return successor
        return root

    从结果来说Java版本的迭代方法的效率不错，而Python版本的递归方法的速度也比较不错，但应该是有更多的方法可以进一步提速的，希望朋友们能够多多指教，非常感谢。