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

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

力扣题目链接(opens new window)

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

class Solution {
private:
    TreeNode* traversal(TreeNode* cur, TreeNode* p, TreeNode* q) {
        if (cur == NULL) return cur;
                                                        // 中
        if (cur->val > p->val && cur->val > q->val) {   // 左
            TreeNode* left = traversal(cur->left, p, q);
            if (left != NULL) {
                return left;
            }
        }

        if (cur->val < p->val && cur->val < q->val) {   // 右
            TreeNode* right = traversal(cur->right, p, q);
            if (right != NULL) {
                return right;
            }
        }
        return cur;
    }
public:
    TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) {
        return traversal(root, p, q);
    }
};

本题是二叉搜索树，按照规则找目标值，只要找到一个比p大比q小的节点返回就欧克

如果cur大于这两个数就向左去找 定义一个left去接 如果找到了left会成为目标值 直接返回left就可以

反之亦然

最后的情况就是符合条件的情况直接return cur就可以

701.二叉搜索树中的插入操作

力扣题目链接(opens new window)

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

class Solution {
public:
    TreeNode* insertIntoBST(TreeNode* root, int val) {
        if (root == NULL) {
            TreeNode* node = new TreeNode(val);
            return node;
        }
        if (root->val > val) root->left = insertIntoBST(root->left, val);
        if (root->val < val) root->right = insertIntoBST(root->right, val);
        return root;
    }
};

终止条件 遇见空节点说明到叶子节点的下一位了 说明找到插入的位置了 创建一个新节点放入值并返回这个节点

然后就是二叉搜索树的逻辑 用叶子节点的左子树或者右子树来接这个节点完成拼接

最后返回根节点

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

力扣题目链接(opens new window)

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

class Solution {
public:
    TreeNode* deleteNode(TreeNode* root, int key) {
        if(root==NULL)return NULL;
        if(root->val==key){
            if(root->left==NULL&&root->right==NULL){
                return NULL;
            }
            else if(root->left!=NULL&&root->right==NULL){
                return root->left;
            }else if(root->right!=NULL&&root->left==NULL){
                return root->right;
            }else{
                TreeNode*cur=root->right;
                while(cur->left!=NULL){
                    cur=cur->left;
                }
                cur->left=root->left;
                return root->right;
            }
        }
        if(root->val>key){
            root->left=deleteNode(root->left,key);
        }if(root->val<key){
            root->right=deleteNode(root->right,key);
        }
        return root;
    }
};

以上代码是自己写的思路代码 没有释放内存

终止条件是找到节点了

分五种情况

第一种是删除的是叶子节点 叶子节点左右都为空 直接返回NULL就行

第二种情况是删除节点 左不为空 右为空 只需要返回要删除节点的的左子树即可（并把要删除的节点释放内存）

第三种情况与第二种逻辑相同 右不为空 左为空 只需要返回要删除节点的的右子树即可

第四种情况是左右都不为空 那么就把要删除节点的左子树 拼接到要删除节点的右子树的最左孩子的左边 然后再返回要删除节点的右子树

第五种情况是没找着要删除的节点 就是第一行

递归条件就是按照二叉搜索树的特性去找将要删除的节点

如果要删除的节点在根节点的左子树里面 就用根节点的左也就是root->left去接递归返回回来的左子树

完整释放内存的代码：

class Solution {
public:
    TreeNode* deleteNode(TreeNode* root, int key) {
        if (root == nullptr) return root; // 第一种情况：没找到删除的节点，遍历到空节点直接返回了
        if (root->val == key) {
            // 第二种情况：左右孩子都为空（叶子节点），直接删除节点， 返回NULL为根节点
            if (root->left == nullptr && root->right == nullptr) {
                ///! 内存释放
                delete root;
                return nullptr;
            }
            // 第三种情况：其左孩子为空，右孩子不为空，删除节点，右孩子补位 ，返回右孩子为根节点
            else if (root->left == nullptr) {
                auto retNode = root->right;
                ///! 内存释放
                delete root;
                return retNode;
            }
            // 第四种情况：其右孩子为空，左孩子不为空，删除节点，左孩子补位，返回左孩子为根节点
            else if (root->right == nullptr) {
                auto retNode = root->left;
                ///! 内存释放
                delete root;
                return retNode;
            }
            // 第五种情况：左右孩子节点都不为空，则将删除节点的左子树放到删除节点的右子树的最左面节点的左孩子的位置
            // 并返回删除节点右孩子为新的根节点。
            else {
                TreeNode* cur = root->right; // 找右子树最左面的节点
                while(cur->left != nullptr) {
                    cur = cur->left;
                }
                cur->left = root->left; // 把要删除的节点（root）左子树放在cur的左孩子的位置
                TreeNode* tmp = root;   // 把root节点保存一下，下面来删除
                root = root->right;     // 返回旧root的右孩子作为新root
                delete tmp;             // 释放节点内存（这里不写也可以，但C++最好手动释放一下吧）
                return root;
            }
        }
        if (root->val > key) root->left = deleteNode(root->left, key);
        if (root->val < key) root->right = deleteNode(root->right, key);
        return root;
    }
};