# Algorithm to convert Binary Search Tree into Balanced Binary Search Tree

In this article, we will explore an algorithm to convert a Binary Search Tree (BST) into a Balanced Binary Search Tree. In a balanced BST, the height of the tree is log N where N is the number of elements in the tree. In the worst case and in an unbalanced BST, the height of the tree can be upto N which makes it same as a linked list. The height depends upon the order of insertion of elements while some other trees like AVL tree has routines to keep their tree balanced which is not present in a normal Binary Search Tree. It is important to keep a BST balanced, as it will give best performance for tasks it is build for like:

• searching elements in O(log N)

The conversion to a Balanced Binary Search Tree takes O(N) time complexity

## Example:

Input of an unbalanced Binary Search Tree:

Output of the same tree but as a balanced Binary Search Tree:

As we know the property of binary search tree, inorder traversal of binary search tree gives element in sorted order which are stored in binary search tree.And then we can form the balanced binary search from the sorted array.

## Algorithm:

1. Traverse given BST in inorder and store result in an array. This step takes O(n) time. Note that this array would be sorted as inorder traversal of BST always produces sorted sequence.
2. Get the Middle of the array and make it root.
3. Recursively do same for left half and right half.
4. Get the middle of left half and make it left child of the root created in step 1.
5. Get the middle of right half and make it right child of the root created in step

## Implementation

Following is the implementation of the above algorithm.

``````#include <bits/stdc++.h>
using namespace std;
struct node
{
int key;
struct node *left, *right;
};

// A utility function to create a new BST node
struct node *newNode(int item)
{
struct node *temp =  (struct node *)malloc(sizeof(struct node));
temp->key = item;
temp->left = temp->right = NULL;
return temp;
}
/* A utility function to insert a new node with given key in BST */
struct node* insert(struct node* node, int key)
{
/* If the tree is empty, return a new node */
if (node == NULL) return newNode(key);

/* Otherwise, recur down the tree */
if (key < node->key)
node->left  = insert(node->left, key);
else if (key > node->key)
node->right = insert(node->right, key);

/* return the (unchanged) node pointer */
return node;
}
/* This function traverse the skewed binary tree and
stores its nodes pointers in vector nodes[] */
void storeBSTNodes(struct node* root, vector<struct node*> &nodes)
{
// Base case
if (root==NULL)
return;

// Store nodes in Inorder (which is sorted
// order for BST)
storeBSTNodes(root->left, nodes);
nodes.push_back(root);
storeBSTNodes(root->right, nodes);
}

/* Recursive function to construct binary tree */
struct node* buildTreeUtil(vector<struct node*> &nodes, int start,
int end)
{
// base case
if (start > end)
return NULL;

/* Get the middle element and make it root */
int mid = (start + end)/2;
struct node *root = nodes[mid];

/* Using index in Inorder traversal, construct
left and right subtress */
root->left  = buildTreeUtil(nodes, start, mid-1);
root->right = buildTreeUtil(nodes, mid+1, end);

return root;
}

// This functions converts an unbalanced BST to
// a balanced BST
struct node* buildTree(struct node* root)
{
// Store nodes of given BST in sorted order
vector<struct node*> nodes;
storeBSTNodes(root, nodes);

// Constucts BST from nodes[]
int n = nodes.size();
return buildTreeUtil(nodes, 0, n-1);
}
/* Function to do preorder traversal of tree */
void preOrder(struct node* node)
{
if (node == NULL)
return;
cout<<node->key<<" ";
preOrder(node->left);
preOrder(node->right);
}

// Driver Program to test above functions
int main()
{
struct node *root = NULL;
root = insert(root, 1);
insert(root, 2);
insert(root, 3);
insert(root, 4);
insert(root, 5);
root = buildTree(root);
cout<<"Pre order Traversal of tree"<<endl;
preOrder(root);
return 0;
}
``````

Output:

``````Pre order Traversal of tree
3 1 2 4 5
``````

## Explanation:

First of all we will do inorder traversal and and store the elements in array.

## Time Complexity:

The Inorder Traversal of Binary search tree in O(n) time complexity.

To form Balanced Binary tree from Sorted array , it takes O(n) time to complete.

Following is the recurrence relation for buildTreeUtil().

``````T(n) = 2T(n/2) + C
T(n) -->  Time taken for an array of size n
C   -->  Constant
(Finding middle of array
linking root to left and right subtrees take constant time)
``````

Therefore, in total this algorithm takes O(N) time to complete.

This is a companion discussion topic for the original entry at http://iq.opengenus.org/algorithm-to-convert-binary-search-tree-into-balanced-binary-search-tree/