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

1. 题目 ：

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

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

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

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

    }

};

2. 题目 ：

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

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

## 递归

class Solution {

private:

    TreeNode* parent;

    void traversal(TreeNode* cur, int val) {

        if (cur == NULL) {

            TreeNode* node = new TreeNode(val);

            if (val > parent->val) parent->right = node;

            else parent->left = node;

            return;

        }

        parent = cur;

        if (cur->val > val) traversal(cur->left, val);

        if (cur->val < val) traversal(cur->right, val);

        return;

    }

public:

    TreeNode* insertIntoBST(TreeNode* root, int val) {

        parent = new TreeNode(0);

        if (root == NULL) {

            root = new TreeNode(val);

        }

        traversal(root, val);

        return root;

    }

};

3. 题目 ：

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

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

首先找到需要删除的节点； 如果找到了，删除它。 说明： 要求算法时间复杂度为 $O(h)$，h 为树的高度。

## 递归

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;

    }

};