代码随想录算法训练营第二十一天|LeetCode235,701,450

代码随想录算法训练营第二十一天|LeetCode235,701,450

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

本题最主要的一点是要想到：利用搜索二叉树有序的性质。从根节点开始搜索，第一个落在[p,q]中间的节点，

    public TreeNode lowestCommonAncestor(TreeNode root, TreeNode p, TreeNode q) {
        if (root == null){
            return null;
        }
        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.二叉搜索树中的插入操作

这个题没有必要去考虑重构二叉搜索树，因此，遍历二叉树即可，如果遇到为null的地方了，就需要加入新节点了。

递归函数的返回值可为TreeNode，

if (root->val > val) root->left = insertIntoBST(root->left, val);
if (root->val < val) root->right = insertIntoBST(root->right, val);
return root;

通过递归函数返回值完成了新加入节点的父子关系赋值操作了，下一层将加入节点返回，本层用root->left或者root->right将其接住。

返回值也可以为空，就需要记录上一个节点。

 public TreeNode insertIntoBST(TreeNode root, int val) {
        //找到需要添加节点的位置了
        if (root == null){
            TreeNode tmp = new TreeNode(val);
            return tmp;
        }

        if (root.val > val){
            root.left = insertIntoBST(root.left,val);
        }else if (root.val < val){
            root.right = insertIntoBST(root.right,val);
        }
        return root;
    }

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

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

说到递归函数的返回值，在二叉树：搜索树中的插入操作 (opens new window)中通过递归返回值来加入新节点， 这里也可以通过递归返回值删除节点。

• 确定终止条件

遇到空返回，其实这也说明没找到删除的节点，遍历到空节点直接返回了

• 确定单层递归的逻辑

这里就把二叉搜索树中删除节点遇到的情况都搞清楚。

有以下五种情况：

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

    public TreeNode deleteNode(TreeNode root, int key) {
        if (root == null){
            return null;
        }

        if (root.val > key){
            root.left = deleteNode(root.left, key);
        }

        if (root.val < key){
            root.right = deleteNode(root.right,key);
        }

        //找到要删除的节点
        if (root.val == key) {
            if (root.left == null && root.right == null) {
                return null;
            }
            //要删除的节点左子树不为空右子树为空
            if (root.left != null && root.right == null) {
                return root.left;
            }
            //要删除的节点左子树为空右子树不为空
            if (root.left == null && root.right != null) {
                return root.right;
            }
            //要删除的节点左右子树都不为空
            if (root.left != null && root.right != null) {
                TreeNode tmp = root.right;
                while (tmp.left != null) {
                    tmp = tmp.left;
                }
                tmp.left = root.left;
                return root.right;
            }

        }
        //要删除的为叶子节点

        return root;
    }