题目
解法
思路:这题的求解依然可以使用到递归的思想,给出一个根节点和key值,删除key值对应的节点,如果当前给出的根节点的val大于key值,那么就可以从把当前根节点的左子节点作为新的根节点,递归调用这个方法;如果根节点的val小于key值,则把当前根节点的右子树当做新的根节点,递归调用这个方法。当然最终可能会找到key值和value值相同的情况,那么这个节点就是需要删除的节点了,然后我采取的思路是找到它的右子树中的最小值,把这个最小值赋给这个节点的val,再删除那个最小值对应的节点。整体思路是这样,但是其中需要考虑到一些null值的情况。
使用递归的思想是解决这类问题速度最快的方式,代码量也可以大大减少。
解题步骤:
1、如果根节点为null直接返回。
2、根节点val大于key值,从左子树出发递归删除;根节点val小于key值,从右子树出发递归删除。
3、根节点等于val的情况。右子树为空,返回左子树;左子树为空,返回右子树;都不为空,找到右子树中的最小值替换被删除的节点。最后返回root节点。
需要注意的是:每次小的递归都需要返回新的根节点,比如递归左子树就需要返回左子节点;如果删除的就是当前节点,那么就要返回更新后的该位置的节点。
完整代码:
class TreeNode {
int val;
TreeNode left;
TreeNode right;
TreeNode(int x) {
val = x;
}
}
class Solution{
/**
* 删除某个节点并返回新的根节点
* @param root
* @param key
* @return
*/
public TreeNode deleteNode(TreeNode root,int key){
if(root==null) //根节点为空,直接返回
return null;
//值大于key,递归以左子节点出发删除;值小于key,递归以右子节点出发删除
if(root.val>key){
root.left=deleteNode(root.left, key);//删除并返回,root.left是新的返回的根节点
}else if(root.val<key){
root.right=deleteNode(root.right, key);
}else{
if(root.right==null){
return root.left;
}
if(root.left==null){
return root.right;
}
TreeNode rightmin=findmin(root.right);//找右子树的最小值来替换这个位置
root.val=rightmin.val;
root.right=deleteNode(root.right, root.val);
}
return root;
}
/**
* 找到最小的节点
* @param root
* @return
*/
private TreeNode findmin(TreeNode root){
while(root.left!=null){
root=findmin(root.left);
}
return root;
}
}