Deleting in a BST

Procedure :

  • Search recursively for the data to be deleted.

  • Call the function on the left subtree if the data to be deleted is less than current node's data.

  • Call the function on the right subtree if the data to be deleted is greater than current node's data.

  • Once the node is found, check the height of it's left and right subtree.

  • If the height of left subtree is greater, find the inorder predecessor and replace the data in node to be deleted with inorder predecessor's data. Now, delete the inorder predecessor as it's a leaf node.

  • If the height of right subtree is greater, find the inorder successor and replace the data in node to be deleted with inorder successor's data. Now, delete the inorder successor as it's a leaf node.

  • If the node to be deleted is a leaf node, check if it is the root node. If yes, make the root as NULL. Then, free the memory.

struct Node *deleteNode(struct Node *ptr, int key) {

	if (ptr == NULL) 
		return NULL;

	// If the node is a leaf node
	if (ptr -> left == NULL && ptr -> right == NULL) {
		// If it's the root node, make root NULL after deletion
		if (ptr == root) 
			root = NULL;
		// Free the memory
		free(ptr);
		return NULL;
	}

	// If value to be deleted is lesser, go to left subtree
	if (key < ptr -> data) 
		ptr -> left = deleteNode(ptr -> left, key);

	// If value to be deleted is greater, go to right subtree
	else if (key > ptr -> data)
		ptr -> right = deleteNode(ptr -> right, key);
	
	// Deleting the node once it's found
	else {
		// Delete from the subtree which has greater height
		if (height(ptr -> left) > height(ptr -> right)) {

			// Find the inorder predecessor for left subtree
			struct Node *inPre = inorderPredecessor(ptr -> left);
			ptr -> data  = inPre -> data;
			ptr -> left = deleteNode(ptr -> left, inPre -> data);
		
		} else {

			// Find the inorder successor for right subtree
			struct Node *inSuc = inorderSuccessor(ptr -> right);
			ptr -> data  = inSuc -> data;
			ptr -> right = deleteNode(ptr -> right, inSuc -> data);
		
		}
	}

	return ptr;

}

Utility Functions :

int height(struct Node *ptr) {

	int x, y;

	if (ptr == NULL)
		return 0;

	x = height(ptr -> left);
	y = height(ptr -> right);
	
	return x > y ? x + 1 : y + 1;

}

struct Node *inorderPredecessor(struct Node *ptr) {

	// Find rightmost child of left subtree
	while (ptr && ptr -> right)
		ptr = ptr -> right;
	return ptr;

}

struct Node *inorderSuccessor(struct Node *ptr) {

	// Find leftmost child of right subtree
	while (ptr && ptr -> left)
		ptr = ptr -> left;
	return ptr;

}

Contributed by Nitin Ranganath

Last updated