LeetCode450——删除二叉搜索树中的节点

Delete Node in a BST

给定一个二叉搜索树的根节点 root 和一个值 key，删除二叉搜索树中的 key 对应的节点，并保证二叉搜索树的性质不变。返回二叉搜索树（有可能被更新）的根节点的引用。

一般来说，删除节点可分为两个步骤：

1. 首先找到需要删除的节点；
2. 如果找到了，删除它。

说明： 要求算法时间复杂度为 O(h)，h 为树的高度。

示例:

root = [5,3,6,2,4,null,7]
key = 3

    5
   / \
  3   6
 / \   \
2   4   7

给定需要删除的节点值是 3，所以我们首先找到 3 这个节点，然后删除它。

一个正确的答案是 [5,4,6,2,null,null,7], 如下图所示。

    5
   / \
  4   6
 /     \
2       7

另一个正确答案是 [5,2,6,null,4,null,7]。

    5
   / \
  2   6
   \   \
    4   7

 

public TreeNode deleteNode(TreeNode root, int key) {
	          
	     if(root==null)
	    	 return null;
	
		   TreeNode node=root;
	        TreeNode lastNode=root;
	            if(key < node.val){
	                //删除节点的值小于当前节点，删除节点在左子树上 
	                root.left=deleteNode(root.left,key);
	            }else if(key > node.val){
	                //删除节点的值大于当前节点，删除的节点在右子树上
	                root.right= deleteNode(root.right,key);             
	            }else{
	                if(root.left ==null || root.right == null){
	                    //无子节点或仅有一个子节点
	                    root= root.left == null ? root.right:root.left;
	                }else{
	                    //同时有左节点和右节点
	                    TreeNode cur = root.right;
	                    while (cur.left !=null) {
	                    //找到右分支的最小值
	                        cur = cur.left;
	                    }
	                root.val = cur.val;
	                root.right = deleteNode(root.right, cur.val);
	                }
	                
	        }
	        return root;
	
	      
		 
	 }