代码随想录 第二十二天 701.二叉搜索树中的插入操作||450.删除二叉搜索树中的节点

 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 对应的节点，并保证二叉搜索树的性质不变。返回二叉搜索树（有可能被更新）的根节点的引用。

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

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

示例:

代码如下：

最关键的一点是，当目标节点左右孩子都不为空的时候，将该节点的左孩子放到该节点的最右侧孩子的左孩子的位置上。

class Solution {
public:
    TreeNode* DeleteElement(TreeNode* root, int val) {
        //情形一：如果没有找到删除的节点，遍历到空节点就直接返回
        if (root == nullptr)return NULL;
        if (root->val == val) {
            if (root->left == nullptr&&root->right!=nullptr) {
                return root->right;
            }
            else if (root->right == nullptr && root->left != nullptr) {
                return root->left;
            }
            else {
                TreeNode* cur = root->right;   //将cur指针放在要被删的元素节点的左子树上
                while (cur->left != nullptr) {
                    cur = cur->left;
                }
                cur->left = root->left;
                TreeNode* tmp = root;
                root = root->right;
                delete tmp;
                return root;
            }
        }
        if (root->val > val) root->left = DeleteElement(root->left, val);
        if (root->val < val) root->right = DeleteElement(root->right, val);
        return root;
    }

};