删除二叉搜索树中的节点
给定一个二叉搜索树的根节点 root 和一个值 key,删除二叉搜索树中的 key 对应的节点,并保证二叉搜索树的性质不变。返回二叉搜索树(有可能被更新)的根节点的引用。
一般来说,删除节点可分为两个步骤:
首先找到需要删除的节点;
如果找到了,删除它。
说明: 要求算法时间复杂度为 O(h),h 为树的高度。
示例:
root = [5,3,6,2,4,null,7]
key = 3
给定需要删除的节点值是 3,所以我们首先找到 3 这个节点,然后删除它。
一个正确的答案是 [5,4,6,2,null,null,7], 如下图所示。
另一个正确答案是 [5,2,6,null,4,null,7]。
解题思路
递归+二叉搜索树的搜索操作
删除操作需要考虑一下三种情况:
1.如果目标节点没有子节点,则可以直接删除目标节点;
2.如果目标节点只有一个子节点,可以用其子节点作为替换;
3.如果目标节点有两个子节点,则需要用其中序序列的后继节点获前驱节点来替换,再删除该目标节点;
需要清除后继节点和前驱节点的概念:
后继节点:代表中序遍历序列的下一个节点,即比当前节点大的最小节点。先取当前节点的右节点,然后一直取该右节点的左节点,直到左节点为空,则最后指向的节点为后继节点。
前驱节点:代表中序遍历序列的前一个节点,即比当前节点小的最大节点。先取当前节点的左节点,然后一直取该左节点的右节点,直到右节点为空,则最后指向的节点为后继节点。
再结合递归搜索操作,可得:
1.如果key>root.val,说明要删除的节点在右子树,root.right = deleteNode(root.right,key)
2.如果key<root.val,说明要删除的节点在左子树,root.left = deleteNode(root.left,key)
3.如果key == root.val,则表示该节点就是我们要删除的节点,则:
-
如果root为叶子节点,则直接删除即可,root=null
-
如果root不是叶子节点且有左子树,则用它的前驱节点的值代替,root.val = predecessor(root)
-
如果root不是叶子节点且有右子树,则用他的后继节点的值代替,root.val = successor(root)
4.返回root
var deleteNode = function(root, key) {
//前驱结点
function predecessor(root){
root = root.left
while(root.right!=null){
root = root.right
}
return root.val
}
//后继结点
function successor(root){
root = root.right
while(root.left!=null){
root = root.left
}
return root.val
}
//删除操作
if(!root){
return root
}
if(key < root.val){
root.left = deleteNode(root.left,key)
}else if(key > root.val){
root.right = deleteNode(root.right,key)
}else{
if(root.left == null && root.right == null){//key为叶子结点
root = null
}else if(root.left != null){//key不为叶子结点且有左子树
root.val = predecessor(root)
root.left = deleteNode(root.left,root.val)
}else if(root.right != null){//key不为叶子结点且有右子树
root.val = successor(root)
root.right = deleteNode(root.right,root.val)
}
}
return root
};