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

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

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

文档讲解：代码随想录

视频讲解：二叉搜索树找祖先就有点不一样了！| 235. 二叉搜索树的最近公共祖先

思路

根据二叉搜索树是有序树的性质，遍历结点，如果结点值介于p和q值之间（可以取等号），则说明该结点为最近公共祖先，否则向二叉树一侧遍历即可。

代码

class Solution {
public:
    TreeNode *lowestCommonAncestor(TreeNode *root, TreeNode *p, TreeNode *q) {
        if (root->val <= p->val && root->val >= q->val || root->val >= p->val && root->val <= q->val)
            return root;

        if (root->val > p->val && root->val > q->val) {
            TreeNode *left = lowestCommonAncestor(root->left, p, q);
            return left;
        }

        if (root->val < p->val && root->val < q->val) {
            TreeNode *right = lowestCommonAncestor(root->right, p, q);
            return right;
        }
        return root;
    }
};

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

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

文档讲解：代码随想录

视频讲解：原来这么简单？ | LeetCode：701.二叉搜索树中的插入操作

思路

思路一： 将二叉搜索树中序遍历转换为有序数组，然后将新值按顺序插入到有序数组中，然后将有序数组依次转换为二叉树的右结点即可。

思路二： 按照新值和二叉搜索树的性质，找到为NULL处插入即可。

代码

思路一代码

class Solution {
public:
    TreeNode *insertIntoBST(TreeNode *root, int val) {
        vec.clear();
        vec.push_back(val);
        traversal(root);
        sort(vec.begin(), vec.end(), cmp);

        root = new TreeNode(vec[0]);
        TreeNode *cur = root;
        for (int i = 1; i < vec.size(); i++) {
            TreeNode * temp = new TreeNode(vec[i]);
            cur->right = temp;
            cur = cur->right;
        }
        return root;
    }

private:
    bool static cmp(const int &a, const int &b) {
        return a < b;
    }

    vector<int> vec;
    void traversal(TreeNode *node) {
        if (node == NULL)
            return;

        vec.push_back(node->val);
        traversal(node->left);
        traversal(node->right);
    }
};

思路二代码

代码随想录方法三代码

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

        return root;
    }
};

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

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

文档讲解：代码随想录

视频讲解：调整二叉树的结构最难！| LeetCode：450.删除二叉搜索树中的节点

思路

分类讨论

情况一： 如果没有找到要删除的结点，直接返回

情况二： 如果要删除的结点为叶子结点，直接删除结点，返回当前结点为NULL

情况三： 如果要删除的结点的左结点（右节点）为空，则删除结点，同时返回右节点（左结点）

情况四： 如果要删除的结点的左右结点都不为空，则删除节点，将左子树放到右子树的左下角，或者将右子树放到左子树的右下角，返回删除节点的右节点或左结点。

代码

class Solution {
public:
    TreeNode *deleteNode(TreeNode *root, int key) {
        if (root == NULL)
            return root;

        if (root->val == key) {
            if (root->left == NULL)
                return root->right;
            else if (root->right == NULL)
                return root->left;
            else {
                TreeNode *cur = root->right;
                while (cur->left)
                    cur = cur->left;
                cur->left = root->left;
                TreeNode *temp = root;
                root = root->right;
                delete temp;
                return root;
            }
        }
        if (root->val > key)
            root->left = deleteNode(root->left, key);
        if (root->val < key)
            root->right = deleteNode(root->right, key);
        return root;
    }
};