/**
 * 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
        TreeNode dummy = new TreeNode(0);
        dummy.left = root;
        TreeNode parent = findNodeParent(dummy, root, value);
        TreeNode target;
        if (parent.left != null && parent.left.val == value) {
            target = parent.left;
        } else if (parent.right != null && parent.right.val == value) {
            target = parent.right;
        } else {
            return dummy.left;
        }
        deleteNode(parent, target);
        return dummy.left;
    }
    
    private TreeNode findNodeParent(TreeNode parent, TreeNode node, int val) {
        if (node == null || node.val == val) {
            return parent;
        }
        if (node.val < val) {
            return findNodeParent(node, node.right, val);
        } else {
            return findNodeParent(node, node.left, val);
        }
    }
    
    private void deleteNode(TreeNode parent, TreeNode target) {
        if (target.right == null) {
            if (parent.right == target) {
                parent.right = target.left;
            } else {
                parent.left = target.left;
            }
        } else {
            // Find first element that is greater than target
            TreeNode node1 = target.right;
            TreeNode node2 = target;
            while (node1.left != null) {
                node2 = node1;
                node1 = node1.left;
            }
            // Remove node1
            if (node1 == node2.left) {
                node2.left = node1.right;
            } else {
                node2.right = node1.right;
            }
            // Remove target
            if (target == parent.right) {
                parent.right = node1;
            } else {
                parent.left = node1;
            }
            node1.left = target.left;
            node1.right = target.right;
        }
    }
}