代码随想录算法训练营Day22|Leetcode235二叉搜索树的最近公共祖先、Leetcode701二叉搜索树中的插入操作、Leetcode450删除二叉搜索树中的节点

Day22打卡！ 时长：2h

今日感想：深刻理解二叉搜索树的应用~

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

题目链接：Leetcode235 二叉搜索树的最近公共祖先

第一想法：普通二叉树的思路是后序遍历，二叉搜索树有什么好处？

讲解后想法： 好处就是只要从上到下去遍历，遇到 cur节点是数值在[p, q]区间中则一定可以说明该节点cur就是q 和 p的公共祖先，而且还是最近的。

递归三部曲

1.确定递归函数返回值以及参数

参数就是当前节点，以及两个结点 p、q。返回值是要返回最近公共祖先TreeNode。

2.确定终止条件

遇到空返回，但根据题目要求不会有空。

3.确定单层递归的逻辑

在遍历二叉搜索树的时候就是寻找区间[p->val, q->val]（左闭右闭）

①如果 cur->val 大于 p->val，同时 cur->val 大于q->val，那么就应该向左遍历（说明目标区间在左子树上）。

（需要注意的是此时不知道p和q谁大，所以两个都要判断。本题是标准的搜索一条边的写法，遇到递归函数的返回值，如果不为空，立刻返回。）

②如果 cur->val 小于 p->val，同时 cur->val 小于 q->val，那么就应该向右遍历（目标区间在右子树）。

③剩下的情况，就是cur节点在区间（p->val <= cur->val && cur->val <= q->val）或者 （q->val <= cur->val && cur->val <= p->val）中，那么cur就是最近公共祖先了，直接返回cur。

难点/细节：合理运用二叉搜索树的特点。

代码：

递归法

class Solution {
    public TreeNode lowestCommonAncestor(TreeNode root, TreeNode p, TreeNode q) {
        if (root.val > p.val && root.val > q.val) return lowestCommonAncestor(root.left, p, q);
        if (root.val < p.val && root.val < q.val) return lowestCommonAncestor(root.right, p, q);
        return root;
    }
}

迭代法

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;
    }
}

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

题目链接：Leetcode701 二叉搜索树中的插入操作

第一想法：这题难道还要先构造出一个二叉搜索树，再进行插入操作吗？

讲解后想法：不能被题目迷惑了，直接遍历插入节点......

递归三部曲

1.确定递归函数参数以及返回值

参数就是根节点和要插入的元素。可以利用返回值完成新加入的节点与其父节点的赋值操作，返回类型为节点类型 。

2.确定终止条件

找到遍历的节点为null的时候，就是要插入节点的位置了，并把插入的节点返回。

3.确定单层递归的逻辑

二叉搜索树不需要遍历整棵树，可以根据插入元素的数值，决定递归方向。下一层将加入节点返回，本层用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 if (root.val > val){
            root.left = 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;
    }
}

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

题目链接：Leetcode450 删除二叉搜索树中的节点 

第一想法：通过遍历删除该节点，如果是中节点就要把右节点替换上来；如果是左节点或右节点直接删除即可。

讲解后想法：有点想简单了。其实有五种情况。

递归三部曲

1.确定递归函数参数以及返回值

可以考虑通过递归返回值删除节点。

2.确定终止条件

遇到空直接返回，也说明没找到删除的节点。

3.确定单层递归的逻辑

有以下五种情况：

• 第一种情况：没找到删除的节点，遍历到空节点直接返回了
• 找到删除的节点
  • 第二种情况：左右孩子都为空（叶子节点），直接删除节点， 返回NULL为根节点
  • 第三种情况：删除节点的左孩子为空，右孩子不为空，删除节点，右孩子补位，返回右孩子为根节点
  • 第四种情况：删除节点的右孩子为空，左孩子不为空，删除节点，左孩子补位，返回左孩子为根节点
  • 第五种情况：左右孩子节点都不为空，则将删除节点的左子树头结点（左孩子）放到删除节点的右子树的最左面节点的左孩子上，返回删除节点右孩子为新的根节点。

难点/细节：第五种情况。

代码：

class Solution {
    public TreeNode deleteNode(TreeNode root, int key) {
        root = delete(root,key);
        return root;
    }

    private TreeNode delete(TreeNode root, int key) {
        if (root == null) return null;

        if (root.val > key) {
            root.left = delete(root.left,key);
        } else if (root.val < key) {
            root.right = delete(root.right,key);
        } else {
            if (root.left == null) return root.right;
            if (root.right == null) return root.left;
            TreeNode tmp = root.right;
            while (tmp.left != null) {
                tmp = tmp.left;
            }
            root.val = tmp.val;
            root.right = delete(root.right,tmp.val);
        }
        return root;
    }
}