二叉搜索树的查找 插入 删除 Java

class Solution {
    public TreeNode searchBST(TreeNode root, int val) {
        if (root == null) {
            return null;
        }
        if (val < root.val) {
            return searchBST(root.left, val);
        }
        else if (val > root.val) {
            return searchBST(root.right, val);
        }
        else {
            return root;
        }
    }
}

class Solution {
    public TreeNode insertIntoBST(TreeNode root, int val) {
        if (root == null) {
            return new TreeNode(val, null, null);
        }
        if (val < root.val) {
            root.left = insertIntoBST(root.left, val);
        }
        else if (val > root.val) {
            root.right = insertIntoBST(root.right, val);
        }
        return root;
    }
}

class Solution {
    public TreeNode deleteNode(TreeNode root, int key) {
        if (root == null) {
            return null;
        }
        // 第一步：先找到要删除的结点
        if (key < root.val) {
            root.left = deleteNode(root.left, key);
        }
        else if (key > root.val) {
            root.right = deleteNode(root.right, key);
        }
        // 如果找到结点
        else {
            // 第一种情况：要被删除的结点只有一个孩子或者没有孩子，返回另一个孩子就行。
            if (root.left == null) {
                return root.right;
            }
            if (root.right == null) {
                return root.left;
            }
            // 第二种情况：如果要删除的结点有两个孩子，则在左子树中找最大的代替，或者在右子树中找最小的代替
            TreeNode max = findMax(root.left); // 先找到左子树中最大的结点
            max.left = deleteMax(root.left);   // 再把这个左子树中最大的结点删除
            max.right = root.right;            // 调整指针的指向
            root = max;
        }
        return root;
    }
	// 左子树中最大的结点肯定没有右孩子，找到最大的结点直接返回它的左孩子就行了
    public TreeNode deleteMax(TreeNode root) {
        if (root.right == null) {
            return root.left;
        }
        root.right = deleteMax(root.right);
        return root;
    }

    public TreeNode findMax(TreeNode root) {
        while (root.right != null) {
            root = root.right;
        }
        return root;
    }
}