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

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

思路：

因为是二叉搜索树，所以公共祖先一定是位于这两个节点区间中的

问题：

为什么第一个是这两个中间的例子，就是公共祖先呢？因为是最先符合要求的，

代码：

TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q)
{
    if (!root) return NULL;
    if (root == p || root == q) return root;

    TreeNode* left = NULL, * right = NULL;
    if (root->left) left = lowestCommonAncestor(root->left, p, q);
    if (root->right) right = lowestCommonAncestor(root->right, p, q);

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

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

思路：

先找到左右两个节点，如果左节点不存在， 右-左， 如果左节点存在 右不存在，那么就 左-右

代码：

TreeNode* insertIntoBST(TreeNode* root, int val) {
    TreeNode* newNode = new TreeNode(val);
    if (!root)return newNode;

    TreeNode* left = NULL, * right = NULL;
    int leftSub = INT_MAX, rightSub = INT_MAX;

    stack<TreeNode*> selected;
    selected.push(root);
    //找出<0最大的节点 和>0最小的节点 也就是他们差值绝对值最小的节点
    while (!selected.empty())
    {
        auto cur_ = selected.top(); selected.pop();

        int val_ = cur_->val - val;
        if (val_ < 0 && abs(leftSub) > abs(val_))
        {
            leftSub = val_;
            left = cur_;
        }
        else if(val_ > 0 && abs(rightSub) > abs(val_))
        {
            rightSub = val_;
            right = cur_;
        }

        if (cur_->left) selected.push(cur_->left);
        if (cur_->right) selected.push(cur_->right);
    }

    //现在需要判断这两个节点， 左节点的右节点，或者右节点的左节点
    
    if (left&&right)
    {
        if (!left->right) left->right = newNode;
        else if (!right->left) right->left = newNode;
    }
    else if (left && !right)
    {
        left->right = newNode;
    }
    else if (!left && right)
    {
        right->left = newNode;
    }
    return root;
}

 最简单的方法，

我们只关注叶子节点，如果当前节点为空，那么 就是目标节点，如果不为空，那么就返回本节点

因为他是一路追寻，所以不会出现给其他节点赋值的情况

代码：

TreeNode* insertIntoBST(TreeNode* root, int val) {
    TreeNode* newNode = new TreeNode(val);
    if (!root)return newNode;
    
    if (root->val < val)
    {
        root->right = insertIntoBST(root->right, val);
    }
    else
    {
        root->left = insertIntoBST(root->left, val);
    }
    return root;
}

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

思路：

找到当前节点后，删掉当前节点，不要想如果我删掉当前节点，那么它的父节点怎么办？

我直接把当前的节点设置为NULL，那么它的父节点指向的也就是NULL，所以不用着急

怎么删去有左右孩子的节点：直接把左子树所有都放到右子树的最左边，就可以了

代码：

TreeNode* deleteNode(TreeNode* root, int key) {
    if (!root) return NULL;

    if (root->val == key)
    {
        if (!root->left && !root->right)
        {
            return NULL;
        }
        if (root->left && !root->right)
        {
            return root->left;
        }
        if (!root->left && root->right)
        {
            return root->right;
        }
        if (root->right && root->left)
        {
            auto 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;
}