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

day22

思路：

新意：一开始以为要记录下前一个结点，这样才好记录下一个结点。

结果可以直接用返回值接住。也就是跳过需要被删除的结点，直接返回下一个结点。

如果没找到需要被删除的元素，return nullptr

需要分情况讨论，如果找到了 需要被删除的元素。

一共有四种。

 

 

 

 注意：

1.释放结点的时候，将需要返回的结点先保存下来。再delete掉当前的根结点

2. 注意在删除左右非空结点的情况下的，while循环条件。

        是 cur->left != nullptr

        如果写成了 cur != nullptr  , 就会导致可能会操作空指针的情况.

        因为无法保证 cur->left 是否为空。之后要操作的是 cur->left。

所以，当需要操作哪个的时候，就需要保证其不为空。

 

class Solution {
public:

    TreeNode* pre = nullptr;
    TreeNode* deleteNode(TreeNode* root, int key) {
        // 觉得和链表一样，应该要有一个结点来指向被删除结点 的前一个结点
        // 问题 ： 如果根结点被删除了应该怎么办？

        if(root == nullptr)
            return nullptr;
        if(root->val == key)
        {
            if(root->left == nullptr && root->right == nullptr)
                return nullptr;
            else if(root->left != nullptr && root->right == nullptr)
            {
                auto node = root->left;    // 保存
                delete root;
                return node;
            }    
            else if(root->left == nullptr && root->right != nullptr)
            {
                auto node = root->right;   // 保存
                delete root;
                return node;
            }               
            else
            {
                TreeNode* cur = root->right;
                while(cur->left != nullptr)   // 注意循环条件
                    cur = cur->left;
                cur->left = root->left;
                TreeNode* temp = root->right;       // 保存需要返回的结点
                delete root;                
                return temp;
            }
        }

        if(key < root->val) root->left = deleteNode(root->left, key);
        if(key > root->val) root->right = deleteNode(root->right, key);

        return root;

    }
};