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

删除结点有三种情况：

1、待删除结点为叶子节点，不存在左右子。
直接删掉就行，返回nullptr回去就是删掉了。

2、待删除结点存在一个子结点
直接将子结点填补到待删除结点的位置即可，即直接返回子结点

3、待删除结点左右子都存在
则有两种删除方法：
》将其右子填补到待删除结点的位置，再将其左子接到右子的右子树中最左子的下方
》将其右子接到左子的左子树的最右处，再将左子填补到待删除结点的位置

先贴上我一遍过的代码🤤 再说问题

class Solution {
public:
    TreeNode* pre = nullptr;//用pre指向前一个结点, 以帮助删除结点
    TreeNode* deleteNode(TreeNode* root, int key) {
        if (root == nullptr) return nullptr;

        if (root->val == key) {//当前结点root就是要删除的结点
            if (pre == nullptr) pre = root;
            //第一种情况
            if (!root->left && !root->right) return nullptr;
            //第二种情况
            else if (!root->left || !root->right) {
                return root->left != nullptr ? root->left : root->right;
            }
            //第三种情况
            else {
                TreeNode* replace = root->right;
                //先将替补填上去
                if (key < pre->val) pre->left = root->right;
                else if (key > pre->val) pre->right = root->right;
                //再将原结点的另一个子树接到替补结点的最左子的左边
                //》先遍历到替补结点的最左子node
                TreeNode* node = replace;
                while(node->left) node = node->left;
                //》再将原结点的另一个子树接到最左子node的左边
                node->left = root->left;
                return replace;
            }
        }
        pre = root;
        if (key < root->val) root->left = deleteNode(root->left, key); 
        else if (key > root->val) root->right = deleteNode(root->right, key); 
        return root;
    }
};

问题就是没有释放内存,因此在删除结点时都要记得添上这两句

delete root;
root==nullptr;