代码随想录|235. 二叉搜索树的最近公共祖先 701.二叉搜索树中的插入操作 450.删除二叉搜索树中的节点

import com.sun.source.doctree.EscapeTree;

import java.security.spec.RSAOtherPrimeInfo;
import java.util.Map;

public class Day17 {
    private static class TreeNode {
        int val;
        TreeNode left;
        TreeNode right;

        public TreeNode(int val) {
            this.val = val;
        }

        public TreeNode(int val, TreeNode left, TreeNode right) {
            this.val = val;
            this.left = left;
            this.right = right;
        }
    }
    //给定一个二叉搜索树, 找到该树中两个指定节点的最近公共祖先。
    //百度百科中最近公共祖先的定义为：“对于有根树 T 的两个结点 p、q，最近公共祖先表示为一个结点 x，
    // 满足 x 是 p、q 的祖先且 x 的深度尽可能大（一个节点也可以是它自己的祖先）。”
    public TreeNode lowestCommonAncestor(TreeNode root, TreeNode p, TreeNode q) {
        //本题就是标准的搜索一条边的写法，遇到递归函数的返回值，如果不为空，立刻返回。
        //二叉搜索树可以依靠val的大小判断递归哪条边
        if(root.val>p.val&&root.val>q.val)
            return lowestCommonAncestor(root.left,p,q);
        else if(root.val<p.val&&root.val<q.val)
            return lowestCommonAncestor(root.right,p,q);
        return root;
    }
    //给定二叉搜索树（BST）的根节点 root 和要插入树中的值 value ，将值插入二叉搜索树。 返回插入后二叉搜索树的根节点。
    // 输入数据 保证 ，新值和原始二叉搜索树中的任意节点值都不同。
    //注意，可能存在多种有效的插入方式，只要树在插入后仍保持为二叉搜索树即可。 你可以返回 任意有效的结果 。
    public TreeNode insertIntoBST(TreeNode root, int val) {
         if(root==null)
             return new TreeNode(val);
         if(val<root.val) root.left= insertIntoBST(root.left,val);
         else root.right= insertIntoBST(root.right,val);
         return root;
    }
//迭代法
    public TreeNode insertIntoBST2(TreeNode root, int val) {
            if (root==null)
                return new TreeNode(val);
            TreeNode newRoot=root;
            while (root!=null){
                if(val>root.val){
                    if(root.right==null)
                        break;
                    root=root.right;
                }
                else {
                    if(root.left==null)
                        break;
                    root=root.left;
                }
            }
            if(val<root.val) root.left= new TreeNode(val);
            else root.right=  new TreeNode(val);
            return newRoot;
    }
    //给定一个二叉搜索树的根节点 root 和一个值 key，删除二叉搜索树中的 key 对应的节点，并保证二叉搜索树的性质不变。
    // 返回二叉搜索树（有可能被更新）的根节点的引用。
    //一般来说，删除节点可分为两个步骤：
    //首先找到需要删除的节点；
    //如果找到了，删除它。
//二叉搜索树删除节点会改变树的结构，将删除情况分开，分别处理
    public TreeNode deleteNode(TreeNode root, int key) {
        if(root==null)
            return null;//没找到
        if(key==root.val){
            //分四种情况
            if(root.left==null&&root.right==null) return null;//叶子节点
            else if(root.right!=null&&root.left==null) return root.right;//只有右子树
            else if(root.left!=null&&root.right==null) return root.left;//只有左子树
            else{//既有左子树又有右子树
                TreeNode cur=root.right;
                while(cur.left!=null){
                    cur=cur.left;//找到右子树的最左节点 当作左子树的根节点
                }
                cur.left=root.left;
                return root.right;
            }
        }
        if (key<root.val)
            root.left=deleteNode(root.left,key);
        if(key>root.val) {
            root.right=deleteNode(root.right,key);
        }
        return root;
    }
}