How to determine if a binary tree is height-balanced? - cook the code

Wednesday 29 November 2017

How to determine if a binary tree is height-balanced?

How to determine if a binary tree is height-balanced?


A tree where no leaf is much farther away from the root than any other leaf. Different balancing schemes allow different definitions of “much farther” and different amounts of work to keep them balanced.
Consider a height-balancing scheme where following conditions should be checked to determine if a binary tree is balanced.
An empty tree is height-balanced. A non-empty binary tree T is balanced if:
1) Left subtree of T is balanced
2) Right subtree of T is balanced
3) The difference between heights of left subtree and right subtree is not more than 1.
The above height-balancing scheme is used in AVL trees. The diagram below shows two trees, one of them is height-balanced and other is not. The second tree is not height-balanced because height of left subtree is 2 more than height of right subtree.


#include <bits/stdc++.h>
using namespace std;

// Data structure to store a Binary Tree node
struct Node
{
int data;
Node *left, *right;
};

// Function to create a new binary tree node having given key
Node* newNode(int key)
{
Node* node = new Node;
node->data = key;
node->left = node->right = nullptr;

return node;
}

// Function to insert given key into the tree
void insert(Node*& root, string level, int key)
{
// tree is empty
if (level.length() == 0)
{
root = newNode(key);
return;
}

int i = 0;
Node* ptr = root;
while (i < level.length() - 1)
{
if (level[i++] == 'L')
ptr = ptr->left;
else
ptr = ptr->right;
}

if (level[i] == 'L')
ptr->left = newNode(key);
else
ptr->right = newNode(key);
}

// Recursive function to check if given binary tree is height balanced or not
int isHeightBalanced(Node* root, bool& isBalanced)
{
// base case: tree is empty
    if (root == nullptr)
return 0;

// get height of left subtree
int left_height = isHeightBalanced(root->left, isBalanced);

// get height of right subtree
int right_height = isHeightBalanced(root->right, isBalanced);

// tree is unbalanced if absolute difference between height of
// its left subtree and right subtree is more than 1
if (abs(left_height - right_height) > 1)
isBalanced = false;

// return height of subtree rooted at current node
return max(left_height, right_height) + 1;
}

// Main function to check if given binary tree is height balanced or not
bool isHeightBalanced(Node* root)
{
bool isBalanced = true;
isHeightBalanced(root, isBalanced);

return isBalanced;
}

// main function
int main()
{
Node* root = nullptr;
/* Construct below tree
  1
/   \
   / \
  2       3
/ \ /
4   5   6   */

    vector<pair<string, int> > keys =
{
        {"", 1}, {"L", 2}, {"R", 3}, {"LL", 4}, {"LR", 5}, {"RL", 6}
};

for (auto pair: keys)
insert(root, pair.first, pair.second);

isHeightBalanced(root)? cout << "Yes": cout << "No";

return 0;
}

2 comments:

Sai Elakiyaa said...

Very useful information on Data Structures, definitely it helps us to protect our site from copied content, if you are Looking for software courses?

SEO Training in Chennai
JAVA Training in Chennai
Big Data Training in Chennai
Selenium Training in Chennai
German Classes in chennai
DOT NET Training in Chennai
Loadrunner Training in Chennai
performance testing training

Janu said...

Nice! you are sharing such helpful and easy to understandable blog. i have no words for say i just say thanks because it is helpful for me



Dot Net Training in Chennai | Dot Net Training in anna nagar | Dot Net Training in omr | Dot Net Training in porur | Dot Net Training in tambaram | Dot Net Training in velachery





Post a Comment