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

力扣：450.删除二叉搜索树中的节点
代码随想录

题目:
给定一个二叉搜索树的根节点 root 和一个值 key，删除二叉搜索树中的 key 对应的节点，并保证二叉搜索树的性质不变。返回二叉搜索树（有可能被更新）的根节点的引用。

一般来说，删除节点可分为两个步骤：
首先找到需要删除的节点；
如果找到了，删除它。

递归思路：
①终止条件：即满足问题的情况：找到了节点值为key的节点。此时要删除节点。怎么删除节点：题目是返回二叉搜索树的根节点，此时即把root视为根节点。
分为四种情况：
1：若只有左节点返回左节点。
2：若只有右节点返回右节点。
3：若左右节点都有那么将左子树移到右子树的最左边，返回右子树。因为右子树都比中间大，中间都比左子树大，所以当两边都有的时候，既要删除中间节点又要保持其为二叉搜索树，所以左子树要移到正确的位置即右子树的最左边。
4：若都没有那么返回空。

②若没有找到根节点，因为是二叉搜索树所以当根节点值大于key删除左子树中满足key的节点，若小于key删除在右子树中满足key的节点。题目是返回二叉搜索树的根节点，给跟的left 或 right赋值即可。

递归代码如下：

class Solution {
public:
    TreeNode* deleteNode(TreeNode* root, int key) {
        if(!root) return nullptr;
        if(root->val == key) {
            if(!root->left)  return root->right;
            else if(!root->right) return root->left;
            else if(root->left && root->right){
                TreeNode * node = root->right;
                while(node->left) {
                    node = node->left;
                }
                node->left = root->left;
                root = root->right;
                return root;
            }
            else return nullptr;
        }
        else if(root->val > key) root->left = deleteNode(root->left,key);
        else root->right = deleteNode(root->right, key);
        return root;
    }
};

普通二叉树的删除节点方法：
基本代码一致。主要是当左右树都有节点时找到大于左子树的值小于右子树的值，这个值就是右子树的最小值（即最左边节点）。它的代码是改变了树的情况将含有key的节点转换为叶子节点，让其再去寻找key的位置。要是我估计就是直接删除然后return。

class Solution {
public:
    TreeNode* deleteNode(TreeNode* root, int key) {
        if (root == nullptr) return root;
        if (root->val == key) {
            if (root->right == nullptr) { // 这里第二次操作目标值：最终删除的作用
                return root->left;
            }
            TreeNode *cur = root->right;
            while (cur->left) {
                cur = cur->left;
            }
            swap(root->val, cur->val); // 这里第一次操作目标值：交换目标值其右子树最左面节点。
        }
        root->left = deleteNode(root->left, key);
        root->right = deleteNode(root->right, key);
        return root;
    }
};

迭代方法：

class Solution {
private:
    // 将目标节点（删除节点）的左子树放到 目标节点的右子树的最左面节点的左孩子位置上
    // 并返回目标节点右孩子为新的根节点
    // 是动画里模拟的过程
    TreeNode* deleteOneNode(TreeNode* target) {
        if (target == nullptr) return target;
        if (target->right == nullptr) return target->left;
        TreeNode* cur = target->right;
        while (cur->left) {
            cur = cur->left;
        }
        cur->left = target->left;
        return target->right;
    }
public:
    TreeNode* deleteNode(TreeNode* root, int key) {
        if (root == nullptr) return root;
        TreeNode* cur = root;
        TreeNode* pre = nullptr; // 记录cur的父节点，用来删除cur
        while (cur) {
            if (cur->val == key) break;
            pre = cur;
            if (cur->val > key) cur = cur->left;
            else cur = cur->right;
        }
        if (pre == nullptr) { // 如果搜索树只有头结点
            return deleteOneNode(cur);
        }
        // pre 要知道是删左孩子还是右孩子
        if (pre->left && pre->left->val == key) {
            pre->left = deleteOneNode(cur);
        }
        if (pre->right && pre->right->val == key) {
            pre->right = deleteOneNode(cur);
        }
        return root;
    }
};