删除BST 的节点
1.题目描述:
给定一个二叉搜索树的根节点 root 和一个值 key,删除二叉搜索树中的 key 对应的节点,并保证二叉搜索树的性质不变。返回二叉搜索树(有可能被更新)的根节点的引用。
2.思路
1.找到需要删除的节点,找到,删除 时间复杂度**O(h)**h为树高 2.调整BST
递归法的涉及单层逻辑
- 确定单层逻辑(主要有五种情况)
找到删除节点,直接返回
- 找到删除节点
- 2.左右孩子为空(叶子节点),直接删除,返回null为根节点
- 3.删除节点的左孩子为空,右孩子不为空,删除节点,右孩子补位,返回右孩子为根节点
- 4.删除右孩子为空,左孩子不为空,删除节点,左孩子补位,返回左孩子为根节点
- 5.左右孩子都不为空,将删除节点左子树头节点,放到删除节点的右子树最左面节点的左孩子上,返回删除节点右孩子为新的根节点
3.代码(递归思路)
严格遵守递归三部曲!!!
package 二叉树;
public class 删除BST的节点 {
/**
* 给定一个二叉搜索树的根节点 root 和一个值 key,删除二叉搜索树中的 key 对应的节点,并保证二叉搜索树的性质不变。返回二叉搜索树(有可能被更新)的根节点的引用。
* 思路:
* 1.找到需要删除的节点,找到,删除 时间复杂度O(h)h为树高
* 2.调整BST
*/
//1.递归法
//确定递归函数参数和返回值
public TreeNode delNode(TreeNode root,int key){
//第一种:没有找到删除节点,直接返回
//确定终止条件为空返回
if(root==null) return root;
/**
* 确定单层逻辑(主要有五种情况)
* 1.未找到删除节点,直接返回
* 找到删除节点
* 2.左右孩子为空(叶子节点),直接删除,返回null为根节点
* 3.删除节点的左孩子为空,右孩子不为空,删除节点,右孩子补位,返回右孩子为根节点
* 4.删除右孩子为空,左孩子不为空,删除节点,左孩子补位,返回左孩子为根节点
* 5.左右孩子都不为空,将删除节点左子树头节点,放到删除节点的右子树最左面节点的左孩子上,返回删除节点右孩子为新的根节点
*/
//2
if(root.val==key){
if(root.left==null&&root.right==null){
return null;
}
//3
else if(root.left==null){
TreeNode retNode=root.right;
return retNode;
}
//4
else if(root.right==null){
TreeNode retNode=root.left;
return retNode;
}
//5
else{
//找右子树最左面节点
TreeNode cur=root.right;
while(cur.left!=null){
cur=cur.left;
}
cur.left=root.left;
TreeNode tmp=root;
root=root.right;
return root;
}
}
if(root.val>key) root.left=delNode(root.left,key);
if(root.val<key) root.right=delNode(root.right,key);
return root;
}
}
完结
|