LeetCode 450. Delete Node in a BST

题目描述

知识点

BST的结点删除

结果

时间复杂度可以的，但是空间复杂度有点多了，但是我好像找不到优化的方法。我就是按照模板写的呀。

实现

码前思考

1. 这是一道典型的关于BST的删除结点的题目，《算法笔记》上有模板，可以直接往这里面去套。。。。。。
2. 这里的BST删除结点与《算法笔记》上的略微不同，这里的函数TreeNode* deleteNode(TreeNode* &root, int key)里面不是传地址的，而是传值，所以我们要设置函数的返回值，从而保证在删除结点之后树的结构的变化。这个函数的意义在于返回删除节点后的子树的根结点。

代码实现

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
 //题中说要时间复杂度为O(H)，其实BTS就满足啦，不用多想
 //题中没有说结点的值是否可以重复？
class Solution {

public:
    //这个函数不是传地址的，所以要有返回值才对，返回删除节点后的子树的根结点
    TreeNode* deleteNode(TreeNode* root, int key) {
        // deleteN(root,key);

        if(root == NULL){   //如果查找失败，那么说明要查找的不存在，返回root即可
            return NULL;
        }else{
            if(root->val == key){//说明找到这个点了
                //如果这个结点没有左右结点
                if(root->left == NULL && root->right == NULL){
                    return NULL;
                }else if(root->left != NULL){
                    TreeNode* pre = findMax(root->left);
                    root->val = pre->val;
                    root->left = deleteNode(root->left,pre->val);
                }else if(root->right != NULL){
                    TreeNode* next = findMin(root->right);
                    root->val = next->val;
                    root->right = deleteNode(root->right,next->val);
                }
            } else if(root->val > key){
                root->left = deleteNode(root->left,key);
            } else{
                root->right = deleteNode(root->right,key);
            }
        }
        return root;
    }

    TreeNode* findMax(TreeNode* root){
        if(root->right == NULL){
            return root; 
        }else{
            return findMax(root->right);
        }
    }

    TreeNode* findMin(TreeNode* root){
        if(root->left == NULL){
            return root;
        }else{
            return findMin(root->left);
        }
    }

    // void deleteN(TreeNode* &root,int key){
    //     if(root == NULL){   //如果查找失败，那么说明要查找的不存在，返回root即可
    //         return;
    //     }else{
    //         if(root->val == key){//说明找到这个点了
    //             //如果这个结点没有左右结点
    //             if(root->left == NULL && root->right == NULL){
    //                 root = NULL;
    //             }else if(root->left != NULL){
    //                 TreeNode* pre = findMax(root->left);
    //                 root->val = pre->val;
    //                 deleteN(root->left,pre->val);
    //             }else if(root->right != NULL){
    //                 TreeNode* next = findMin(root->right);
    //                 root->val = next->val;
    //                 deleteN(root->right,next->val);
    //             }
    //         } else if(root->val > key){
    //             deleteN(root->left,key);
    //         } else{
    //             deleteN(root->right,key);
    //         }
    //     }
    // }
};

码后思考

1. 感觉我自己对BST删除结点的操作还是有很多不理解的地方，可能是我对递归的思想掌握的不好造成的吧😔
2. 要明确删除一个结点有两大步——“查找”和“删除”；
3. 删除分为三种情况：① 没有左右子树；② 有左子树；③ 没有左子树，有右子树。
4. 针对删除的三种情况，可以采取不同的措施。其中第二种和第三种中涉及到了递归的调用函数。关于这里的递归，我们只要想着框架的思想，不要想递归的细节，不然很难绕的。。。
5. TreeNode* deleteNode(TreeNode* root, int key)的含义是在以root为根结点的子树中删除key。一定要把我这个含义！！！