二叉搜索树插入和删除节点

701 二叉搜索树中的插入操作

给定二叉搜索树（BST）的根节点 root 和要插入树中的值 value ，将值插入二叉搜索树。 返回插入后二叉搜索树的根节点。 输入数据 保证 ，新值和原始二叉搜索树中的任意节点值都不同。

注意，可能存在多种有效的插入方式，只要树在插入后仍保持为二叉搜索树即可。 你可以返回 任意有效的结果 。

思路一：我的版本比较粗暴，定义了一个dfs函数对搜索树进行遍历，寻找插入位置：当root节点的值小于val，判断root.right是否为空，若为空则插入节点；当root节点的值大于val，判断root.left是否为空，若为空则插入，否则就继续遍历。

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode() {}
 *     TreeNode(int val) { this.val = val; }
 *     TreeNode(int val, TreeNode left, TreeNode right) {
 *         this.val = val;
 *         this.left = left;
 *         this.right = right;
 *     }
 * }
 */
class Solution {
    public TreeNode insertIntoBST(TreeNode root, int val) {
        if(root==null) return new TreeNode(val);
        dfs(root,val);
        return root;
    }
    public void dfs(TreeNode root,int val){
        if(root.val<val){
            if(root.right==null){
                root.right = new TreeNode(val);
                return;
            }
            insertIntoBST(root.right,val);
        }else{
            if(root.left==null){
                root.left = new TreeNode(val);
                return;
            }
            insertIntoBST(root.left,val);
        }
    }
}

思路二：我的版本递归函数没有返回值，没有利用父子节点之间的传递来解题，我们可以让下一层将加入节点返回，本层用root.left或root.right接住：

class Solution {
    public TreeNode insertIntoBST(TreeNode root, int val) {
        if (root == null) // 如果当前节点为空，也就意味着val找到了合适的位置，此时创建节点直接返回。
            return new TreeNode(val);
            
        if (root.val < val){
            root.right = insertIntoBST(root.right, val); // 递归创建右子树
        }else{
            root.left = insertIntoBST(root.left, val); // 递归创建左子树
        }
        return root;
    }
}

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

给定一个二叉搜索树的根节点 root 和一个值 key，删除二叉搜索树中的 key 对应的节点，并保证二叉搜索树的性质不变。返回二叉搜索树（有可能被更新）的根节点的引用。

一般来说，删除节点可分为两个步骤：

1 首先找到需要删除的节点；
2 如果找到了，删除它。

思路：写递归函数，利用二叉搜索树的性质，判断root.val和key的大小进行搜索，寻找删除节点。找到之后分两种情况，若root的left或right有一方为空，则直接返回另一方，与父节点链接即可；若root的left和right都不为空，则寻找到root的right的最左下角元素node，将node.lelft与root.left链接，并将root.right替换root。

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode() {}
 *     TreeNode(int val) { this.val = val; }
 *     TreeNode(int val, TreeNode left, TreeNode right) {
 *         this.val = val;
 *         this.left = left;
 *         this.right = right;
 *     }
 * }
 */
class Solution {
    public TreeNode deleteNode(TreeNode root, int key) {
        if(root==null) return root;
        if(key>root.val){
            root.right = deleteNode(root.right,key);
        }else if(key<root.val){
            root.left = deleteNode(root.left,key);
        }else{
            if(root.left==null) return root.right;
            if(root.right==null) return root.left;
            TreeNode node = root.right;
            while(node.left!=null){
                node = node.left;
            }
            node.left = root.left;
            root = root.right;
        }
        return root;
    }
    
}