算法-二叉查找树-删除节点

题目：给定一棵具有不同节点值的二叉查找树，删除树中与给定值相同的节点。如果树中没有相同值的节点，就不做任何处理。你应该保证处理之后的树仍是二叉查找树。

样例：

给出如下二叉查找树：
          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);

    }

}