代码随想录第21天 | 二叉搜索树的最近公共祖先 、二叉搜索树中的插入操作 、删除二叉搜索树中的节点

一、前言

参考文献：代码随想录

这是第八天二叉树了，感觉对自我的提升还是有很多的，但是还需多多刷题，掌握二叉树技巧。

二、搜索二叉树最进的公共祖先

1、思路：

关于最近公共祖先，我第一个想到的还是昨天二叉树的最近公共祖先，顺便复习一下。

代码如下：

class Solution {
public:
    TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) {
        if (root == NULL || root == p || root == q) return root;
        // 左
        TreeNode* left =  lowestCommonAncestor(root->left, p, q);

        // 右
        TreeNode* right = lowestCommonAncestor(root->right, p ,q);

        // 中
        if (right == NULL && left != NULL) return left;
        else if (right != NULL && left == NULL) return right;
        else if (right != NULL && left != NULL) return root;
        else return NULL;
        
    }
};

但是，在这里我们可以利用二叉搜索树的性质，让代码变得更加简单，我总结了一下几点：

（1）返回值还是TreeNode类型，传入参数还是那几个；

（2）终止条件依旧为

if (root == NULL) return root;

 （3）采用的是后序遍历，主要判断就是所遍历的值是否在p，q这个区间，在这个区间出现的第一个几点就为这个公共节点，理解这个就是解题的关键，在小于这个区间，就往左子树遍历，大于就往右子树遍历，这就是核心思路，这里一定要返回各个函数，因为公共祖先需要传递到根节点返回

2、代码如下：

class Solution {
private:
    TreeNode* getAncestor(TreeNode* root, TreeNode* p, TreeNode* q) {
        if (root == NULL) return root;
        // 左
        if (p->val < root->val && q->val < root->val) {
            // 适当的返回，传递公共祖先
            return getAncestor(root->left, p , q);
        }
        // 右
        else if (p->val > root->val && q->val > root->val) {
            return getAncestor(root->right, p ,q);
        }
        // 中
        else {
            return root;
        }
    }
public:
    TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) {
        return getAncestor(root, p , q);
    }
};

三、二叉树中的插入操作

1、思路:

我最开始的想法是找到了最底层的第一个叶子节点，然后接入就行了，这是迭代法，这不是今天的主题，接着就是学了卡哥和GPT教授的思路，写出了一个递归法，有两个小版本，第一个比较繁琐：

（1）返回值为TreeNode，传入参数就为root和val

（2）遍历到NULL时，创建一个新节点返回给父节点

（3）在BST树中并不需要遍历全部的节点，只需要用过比较大小关系来找到适合插入的叶子节点

（4）一定要有root->left (root->right)去接住返回的新节点，不然就连接不成功了。

（5）返回root

2、代码如下：

class Solution {
public:
    TreeNode* insertIntoBST(TreeNode* root, int val) {
        if (root == NULL) {
            return new TreeNode(val);
        }

        if (val < root->val) {
            if (root->left == NULL) {
                root->left = new TreeNode(val);
            } else {
                root->left = insertIntoBST(root->left, val);
            }
        } else {
            if (root->right == NULL) {
                root->right = new TreeNode(val);
            } else {
                root->right = insertIntoBST(root->right, val);
            }
        }

        return root;
    }
};

这里其中的很多细节可以省略，简洁版如下：

class Solution {
public:
    TreeNode* insertIntoBST(TreeNode* root, int val) {
        if (root == NULL) {
            // 添加一个节点，返回上一层，就完成了父子结点的操作了
            TreeNode* kk = new TreeNode(val);
            return kk;
        }
        // 找到适合元素插入的位置
        if (root->val >val) {
            // 接住节点
            root->left = insertIntoBST(root->left, val);

        } else {
            // 接住节点
            root->right = insertIntoBST(root->right, val);

        }
        return root;
    }
};

四、删除二叉搜索树中的节点

1、思路：

这道题分为五种情况，

（1）删除叶子节点，左右都为空；

（2）左不为空，右为空；

（3）右为空，左不为空；

（4）左不为空，右不为空（需要大幅度调整树的结构，如果以删除节点的右子树来顶替的话，就需要找右边最小的节点来接被删除结点的左子树；

（5）没有找到；

采用的是递归，采用前序遍历

2、代码如下：

class Solution {
public:
    TreeNode* deleteNode(TreeNode* root, int key) {
        // 情况一：没找到节点
        if (root == NULL) return root;
        if (root->val == key) {
            // 情况二：找到的是叶子节点
            if (root->left == nullptr && root->right == nullptr) {
                delete root;
                return nullptr;
            }
            // 情况三：左不为空，右为空
            else if (root->left == nullptr) {
                auto temp = root->right;
                delete root;
                return temp;
            }
            // 情况四：右为空，左不为空
            else if (root->right == nullptr) {
                auto temp = root->left;
                delete root;
                return temp;
            }
            // 情况四：左右都不为空
            else{
                // 从右子树找
                auto cur = root->right;
                // 找右子树最左边的值，这个值最小，接住root左子树
                while (cur->left != nullptr) {
                    cur = cur->left;
                }
                // 把左子树接到右子树的最左边
                cur->left = root->left;
                // 方便删除
                TreeNode* temp = root;
                // 连接右子树
                root = root->right;
                delete temp;
                return root;
            }
        }
        // 左遍历，用root接住
        if (key < root->val) root->left = deleteNode(root->left, key);
        if (key > root->val) root->right = deleteNode(root->right, key);
        return root;
    }
};

学习时间：2小时

leave message:

Courage is like love;it must have hope for nourishment.

勇气就像爱，需要希望来滋养。