题目:给定一棵具有不同节点值的二叉查找树,删除树中与给定值相同的节点。如果树中没有相同值的节点,就不做任何处理。你应该保证处理之后的树仍是二叉查找树。
样例:
给出如下二叉查找树:
5
/ \
3 6
/ \
2 4
删除节点3之后,你可以返回:
5
/ \
2 6
\
4
或者:
5
/ \
4 6
/
2
代码如下:
当要删除的结点存在左右子树,首先要找到该节点的后继节点(左子树中最大的/右子树中最小的)
/**
* Definition of TreeNode:
* public class TreeNode {
* public int val;
* public TreeNode left, right;
* public TreeNode(int val) {
* this.val = val;
* this.left = this.right = null;
* }
* }
*/
public class Solution {
/*
* @param root: The root of the binary search tree.
* @param value: Remove the node with given value.
* @return: The root of the binary search tree after removal.
*/
public TreeNode removeNode(TreeNode root, int value) {
// write your code here
if(root == null)
return root;
if(value > root.val)
root.right = removeNode(root.right,value);
else if(value < root.val)
root.left = removeNode(root.left,value);
else if(root.left != null && root.right != null) //上述情况
{
root.val = findMin(root.right).val;
root.right = removeNode(root.right,root.val);
}
else
root = (root.left != null) ? root.left :root.right;// 三种情况:左子树非空,右子树空;左子树空,右子树非空;左子树空,右子树空。
return root;
}
public TreeNode findMin(TreeNode t){
if(t == null)
return t;
else if(t.left == null)
return t;
return findMin(t.left);
}
}