删除二叉搜索树的节点

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

题目链接

给定一个二叉搜索树的根节点 root 和一个值 key，删除二叉搜索树中的 key 对应的节点，并保证二叉搜索树的性质不变。返回二叉搜索树（有可能被更新）的根节点的引用。
一般来说，删除节点可分为两个步骤：
首先找到需要删除的节点；如果找到了，删除它。说明：要求算法时间复杂度为 O(h)，h 为树的高度。

示例:

递归三部曲

1.返回值和参数：输入根节点和删除的节点值，返回删除后的根节点
2.单层递归逻辑：找到删除的点，然后将其左子树放在其右子树的下面，返回右子树的根节点
3.终止条件：节点为空

    TreeNode* deleteNode(TreeNode* root, int key) {
        if(root == nullptr)
            return root;
        if(root->val == key)
        {
            //删除节点为叶子节点
            if(!root->left && !root->right) return nullptr;
                
            //右子树为空，则左子树为根节点
            if(root->right == nullptr) return  root->left;
                
            //左子树为空，则返回右子树
            if(root->left == nullptr) return root->right;
                
            //左右子树都不为空，将左子树放到右子树的最左边，返回右子树的根节点
            TreeNode* rightNode = root->right;
            while(rightNode->left)
                rightNode = rightNode->left;
            rightNode->left = root->left; 
            root->left = nullptr;
            return root->right;  
        }
        if(key > root->val)
            root->right = deleteNode(root->right, key);
        else
            root->left = deleteNode(root->left, key);
        return root;
    }