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

235

题目：

给定一个二叉搜索树, 找到该树中两个指定节点的最近公共祖先。

百度百科中最近公共祖先的定义为：“对于有根树 T 的两个结点 p、q，最近公共祖先表示为一个结点 x，满足 x 是 p、q 的祖先且 x 的深度尽可能大（一个节点也可以是它自己的祖先）。”

例如，给定如下二叉搜索树:  root = [6,2,8,0,4,7,9,null,null,3,5]

题目链接：235. 二叉搜索树的最近公共祖先 - 力扣（LeetCode）

思路：

第一种，递归，可使用基于普通二叉树，也可基于二叉搜索树

第二种，基于二叉搜索树迭代

详情：

代码随想录 (programmercarl.com)

代码：

//迭代
class Solution {
    public TreeNode lowestCommonAncestor(TreeNode root, TreeNode p, TreeNode q) {
        while (true) {
            if (root.val > p.val && root.val > q.val) {
                root = root.left;
            } else if (root.val < p.val && root.val < q.val) {
                root = root.right;
            } else {
                break;
            }
        }
        return root;
    }
}
//普通二叉树既可以
class Solution {
    public TreeNode lowestCommonAncestor(TreeNode root, TreeNode p, TreeNode q) {
       if(root==null)   return null;
       if(root==p||root==q) return root;
       TreeNode left =  lowestCommonAncestor(root.left,p,q);
       TreeNode right = lowestCommonAncestor(root.right,p,q);
       if(left!=null && right!=null)    return root;
       else if(left!=null && right==null)   return left;
       else if(right!=null && left==null)   return right;
       else return null;
    }
}
//利用二叉搜索树
class Solution {
    public TreeNode lowestCommonAncestor(TreeNode root, TreeNode p, TreeNode q) {
        if(root==null)  return null;
        if((root.val<=p.val && root.val>=q.val)||(root.val>=p.val && root.val<=q.val))  return root;
        TreeNode left =  lowestCommonAncestor(root.left,p,q);
        TreeNode right = lowestCommonAncestor(root.right,p,q);
        if(left!=null)  return left;
        if(right!=null) return right;
        return null;
    }
}

701

题目：

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

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

题目链接：701. 二叉搜索树中的插入操作 - 力扣（LeetCode）

思路：

第一种，递归法，找到符合的空结点就可以赋值。

第二种，迭代法，找到合适的位置就赋值。

详情：代码随想录 (programmercarl.com)

代码：

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

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

        return newRoot;
    }
}

450

题目：

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

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

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

题目链接： 450. 删除二叉搜索树中的节点 - 力扣（LeetCode）

思路：

删除有两种方式，一种是将合适的值赋给该结点，一种是将指向改变，同样的，方法既可以递归也可以迭代，但是注意当该结点的左右孩子都不为空需要特殊处理。

详情：代码随想录 (programmercarl.com)

代码：

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