前驱结点: 先取当前节点的左节点,取该节点的最右节点
后继结点: 先取当前节点的右节点,取该节点的最左节点
算法:
如果 key > root.val,说明要删除的节点在右子树,root.right = deleteNode(root.right, key)。
如果 key < root.val,说明要删除的节点在左子树,root.left = deleteNode(root.left, key)。
如果 key == root.val,则该节点就是我们要删除的节点,则:
①:如果该节点是叶子节点,则直接删除它:root = null。
②:如果该节点不是叶子节点且有右节点,则用它的后继节点的值替代 root.val = successor.val,然后删除后继节点。
③:如果该节点不是叶子节点且只有左节点,则用它的前驱节点的值替代 root.val = predecessor.val,然后删除前驱节点
(为什么不直接返回root.left,因为这时候的root可能是子树,所以连接到root.left,
但是不是返回root.left,而是返回根节点)
④:返回 root。
class Solution {
public TreeNode deleteNode(TreeNode root, int key) {
if(root==null){
return null;
}
if(root.val<key){
root.right=deleteNode(root.right,key);
}
if(root.val>key){
root.left=deleteNode(root.left,key);
}
if(root.val==key){
if(root.left==null&&root.right==null){
root=null;
}
else if(root.right!=null){
root.val=laternode(root);
root.right=deleteNode(root.right,root.val);
}
else{
root.val=prenode(root);
root.left=deleteNode(root.left,root.val);
}
}
return root;
}
int prenode(TreeNode root){
TreeNode node=root.left;
while(node.right!=null){
node=node.right;
}
return node.val;
}
int laternode(TreeNode root){
TreeNode node=root.right;
while(node.left!=null){
node=node.left;
}
return node.val;
}
}
lass Solution {
public TreeNode insertIntoBST(TreeNode root, int val) {
if(root==null){
root=new TreeNode(val);
}
if(root.val>val){
root.left=insertIntoBST(root.left,val);
}
if(root.val<val){
root.right=insertIntoBST(root.right,val);
}
return root;
}
}