Leetcode 450. Delete Node in a BST(删除BST中的节点)

题目

Leetcode 450. Delete Node in a BST(删除BST中的节点)

解法

思路:这题的求解依然可以使用到递归的思想,给出一个根节点和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;
	}	
}