Thursday, May 5, 2022

Question 70 : Can you write algorithm to delete a node in binary search tree

 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.


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.


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:


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 node.

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:


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

Java program to delete node in Binary search tree


public class BinarySearchTreeDeleteMain { 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 root.right = 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

Don't miss the next article! 
Be the first to be notified when a new article or Kubernetes experiment is published.                            

 

 Share This

You may also like

Kubernetes Microservices
Python AI/ML
Spring Framework Spring Boot
Core Java Java Coding Question
Maven AWS