How to delete a node from binary search tree in java

If you want to practice data structure and algorithm programs, you can go through data structure and algorithm interview questions.
In this post, we will see how to delete a node from binary search tree.
There are two parts to it.
  • Search the node
  • After searching that node, delete the node.
There are three cases which we may need to consider while deleting a node from binary search tree.
  • If node has no child
  • If node has one child
  • If node has two children.

If node has no child

It is pretty straight forward case. We need to search the node and make it null.
Delete node of Binary Search tree with no child

If node has one children

If node have one children then we need to connect parent of removed node directly to child of the removed node.
Delete node of Binary Search tree with one child

If node has two children

It is complicated case. If it has two nodes, we need to connect parent of node to the leftmost node(minimum) of right sub tree or rightmost node(maximum) of left subtree.

Lets understand this case with example:

Delete node of Binary Search tree with two child
As you can see, we are replacing node 40 with node 50. Node 50 is minimum element in right subtree of 40 and deleting node 50 after moving as there will duplicate nodes.

Why minimum element of right sub tree?

Here we are using binary search tree property that tree can be represented in multiple ways.

For example:
Binary Search tree representation

We are using same property while deleting nodes with two children.

Complete java program:

package org.arpit.java2blog;

public class BinarySearchTreeMain {
 public static class TreeNode {
  int data;
  TreeNode left;
  TreeNode right;

  TreeNode(int data) {
   this.data = data;
  }
 }

 // Get minimum element in binary search tree
 public static TreeNode minimumElement(TreeNode root) {
  if (root.left == null)
   return root;
  else {
   return minimumElement(root.left);
  }
 }

 public static TreeNode deleteNode(TreeNode root, int value) {
  if (root == null)
   return null;
  if (root.data > value) {
   root.left = deleteNode(root.left, value);
  } else if (root.data < value) {
   root.right = deleteNode(root.right, value);

  } else {
   // if nodeToBeDeleted have both children
   if (root.left != null && root.right != null) {
    TreeNode temp = root;
    // Finding minimum element from right
    TreeNode minNodeForRight = minimumElement(temp.right);
    // Replacing current node with minimum node from right subtree
    root.data = minNodeForRight.data;
    // Deleting minimum node from right now
    deleteNode(root.right, minNodeForRight.data);

   }
   // if nodeToBeDeleted has only left child
   else if (root.left != null) {
    root = root.left;
   }
   // if nodeToBeDeleted has only right child
   else if (root.right != null) {
    root = root.right;
   }
   // if nodeToBeDeleted do not have child (Leaf node)
   else
    root = null;
  }
  return root;
 }

 public static TreeNode insert(TreeNode root, TreeNode nodeToBeInserted) {
  if (root == null) {
   root = nodeToBeInserted;
   return root;
  }

  if (root.data > nodeToBeInserted.data) {
   if (root.left == null)
    root.left = nodeToBeInserted;
   else
    insert(root.left, nodeToBeInserted);
  } else if (root.data < nodeToBeInserted.data)
   if (root.right == null)
    root.right = nodeToBeInserted;
   else
    insert(root.right, nodeToBeInserted);
  return root;
 }

 public static void inOrder(TreeNode root) {
  if (root == null)
   return;
  inOrder(root.left);
  System.out.print(root.data + " ");
  inOrder(root.right);
 }

 public static void main(String[] args) {

  // Creating a binary search tree
  TreeNode rootNode = createBinarySearchTree();

  System.out.println("Binary tree:");
  inOrder(rootNode);
  System.out.println();
  System.out.println("Deleting node 40 which have two children:");
  TreeNode rootNodeRes = deleteNode(rootNode, 40);
  inOrder(rootNodeRes);
 }

 public static TreeNode createBinarySearchTree() {
  TreeNode rootNode = new TreeNode(40);
  TreeNode node20 = new TreeNode(20);
  TreeNode node10 = new TreeNode(10);
  TreeNode node30 = new TreeNode(30);
  TreeNode node60 = new TreeNode(60);
  TreeNode node50 = new TreeNode(50);
  TreeNode node70 = new TreeNode(70);
  TreeNode node5 = new TreeNode(5);
  TreeNode node13 = new TreeNode(13);
  TreeNode node55 = new TreeNode(55);

  insert(null, rootNode);
  insert(rootNode, node20);
  insert(rootNode, node10);
  insert(rootNode, node30);
  insert(rootNode, node60);
  insert(rootNode, node50);
  insert(rootNode, node70);
  insert(rootNode, node5);
  insert(rootNode, node13);
  insert(rootNode, node55);
  return rootNode;
 }
}

When you run above program, you will get following output:
Binary tree:
5 10 13 20 30 40 50 55 60 70 
Deleting node 40 which have two children:
5 10 13 20 30 50 55 60 70 


Written by Arpit:

If you have read the post and liked it. Please connect with me on Facebook | Twitter | Google Plus

 

Java tutorial for beginners Copyright © 2012