Leetcode 450. 删除二叉搜索树中的节点 C++

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

题目

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

一般来说，删除节点可分为两个步骤：

首先找到需要删除的节点；
如果找到了，删除它。

说明：

要求算法时间复杂度为 O(h)，h 为树的高度。

示例:

root = [5,3,6,2,4,null,7]
key = 3

    5
   / \
  3   6
 / \   \
2   4   7

给定需要删除的节点值是 3，所以我们首先找到 3 这个节点，然后删除它。

一个正确的答案是 [5,4,6,2,null,null,7], 如下图所示。

    5
   / \
  4   6
 /     \
2       7

另一个正确答案是 [5,2,6,null,4,null,7]。

    5
   / \
  2   6
   \   \
    4   7

题解

因为是二叉搜索树，我们可以根据二叉搜索树的性质找到要删除的结点，同时我们需要记录一下它的父节点。
如果不存在该结点，则不需要操作。
如果是根节点，

• 若左子树为空，则直接返回右子树
• 若左子树不为空，则把根节点的右子树放到左子树的最右部分

其他情况，和上种情况类似，只是需要把删除结点的父节点和下面的结点进行连接
详细过程见代码

代码

TreeNode* deleteNode(TreeNode* root, int key) {
        TreeNode *pre=root,*now = root;
        while(now && now->val != key){
            pre = now;
            if(now->val < key)  now = now->right;
            else    now = now->left;
        }
        if(now == NULL) return root;		//没有找到该结点
        else if(now == root){				//删除的是根节点
            TreeNode* newL = now->left;
            if(newL == NULL){				//直接返回右子树
                if(now->right == NULL)  return nullptr;
                else root = now->right;
            }else{							//把右子树放到左子树的最右边
                while(newL->right) newL = newL->right;
                newL->right = now->right;
                root = root->left;
            }
        }else{
            TreeNode* newL = now->left;
            if(newL == NULL){			//删除结点无左子树
                if(pre->val < key){		//如果删除结点的父节点小于删除结点，那么删除结点的右子树应该放在父节点的右子树上
                    if(now->right == NULL)  pre->right = NULL;
                    else    pre->right = now->right;
                }else{				//如果删除结点的父节点大于删除结点，那么删除结点的右子树应该放在父节点的左子树上
                    if(now->right == NULL)  pre->left = NULL;
                    else    pre->left = now->right;
                }
            }else{		//先找到左子树的最右部分
                while(newL->right)  newL = newL->right;
                newL->right = now->right;		//将右子树放到左子树的最右部分
                if(pre->val < key)  pre->right = now->left;		//如果删除结点的父节点小于删除结点，那么新子树的根节点是父节点的右子树上
                else    pre->left = now->left;		//如果删除结点的父节点大于删除结点，那么新子树的根节点是父节点的左子树上
            }
        }
        return root;
    }

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/delete-node-in-a-bst
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。