代码随想录算法训练营第22天|235. 二叉搜索树的最近公共祖先 、701.二叉搜索树中的插入操作、450.删除二叉搜索树中的节点

今日内容： 

●  235. 二叉搜索树的最近公共祖先 

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

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

235. 二叉搜索树的最近公共祖先

class Solution {
    public TreeNode lowestCommonAncestor(TreeNode root, TreeNode p, TreeNode q) {
        //从上往下遍历，第一次遇到[p,q]区间的节点值，那么就是bst的最近公共祖先
        //只需要搜索一条边
        if(root == null)return root;
        if(root.val > p.val && root.val > q.val) {
            TreeNode left = lowestCommonAncestor(root.left, p, q);
            if(left != null) {
                return left;
            }
        }
        if(root.val < p.val && root.val < q.val) {
            TreeNode right = lowestCommonAncestor(root.right, p, q);
            if(right != null) {
                return right;
            }
        }
        return root;
    }
}

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

class Solution {
    public TreeNode insertIntoBST(TreeNode root, int val) {
        //bst不需要遍历整棵树
        //用返回值赋值来实现父子节点的关系
        if(root == null) {
            TreeNode node = new TreeNode(val);
            return node;
        }
        if(root.val > val)root.left = insertIntoBST(root.left, val);
        if(root.val < val)root.right = insertIntoBST(root.right, val);
        return root;
    }
}

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

class Solution {
  public TreeNode deleteNode(TreeNode root, int key) {
    if (root == null) return root;
    if (root.val == key) {
      if (root.left == null) {
        return root.right;
      } else if (root.right == null) {
        return root.left;
      } else {
        TreeNode cur = root.right;
        while (cur.left != null) {
          cur = cur.left;
        }
        cur.left = root.left;
        root = root.right;
        return root;
      }
    }
    if (root.val > key) root.left = deleteNode(root.left, key);
    if (root.val < key) root.right = deleteNode(root.right, key);
    return root;
  }
}