Maximum width of binary tree

// C++ program to calculate width of binary tree
#include <bits/stdc++.h>
using namespace std;

/* A binary tree node has data, pointer to left child
and a pointer to right child */
class node {
public:
	int data;
	node* left;
	node* right;
};

/*Function prototypes*/
int getWidth(node* root, int level);
int height(node* node);
node* newNode(int data);

/* Function to get the maximum width of a binary tree*/
int getMaxWidth(node* root)
{
	int maxWidth = 0;
	int width;
	int h = height(root);
	int i;

	/* Get width of each level and compare
		the width with maximum width so far */
	for (i = 1; i <= h; i++) {
		width = getWidth(root, i);
		if (width > maxWidth)
			maxWidth = width;
	}

	return maxWidth;
}

/* Get width of a given level */
int getWidth(node* root, int level)
{

	if (root == NULL)
		return 0;

	if (level == 1)
		return 1;

	else if (level > 1)
		return getWidth(root->left, level - 1)
			+ getWidth(root->right, level - 1);
}

/* UTILITY FUNCTIONS */
/* Compute the "height" of a tree -- the number of
	nodes along the longest path from the root node
	down to the farthest leaf node.*/
int height(node* node)
{
	if (node == NULL)
		return 0;
	else {
		/* compute the height of each subtree */
		int lHeight = height(node->left);
		int rHeight = height(node->right);
		/* use the larger one */

		return (lHeight > rHeight) ? (lHeight + 1)
								: (rHeight + 1);
	}
}

/* Helper function that allocates a new node with the
given data and NULL left and right pointers. */
node* newNode(int data)
{
	node* Node = new node();
	Node->data = data;
	Node->left = NULL;
	Node->right = NULL;
	return (Node);
}

/* Driver code*/
int main()
{
	node* root = newNode(1);
	root->left = newNode(2);
	root->right = newNode(3);
	root->left->left = newNode(4);
	root->left->right = newNode(5);
	root->right->right = newNode(8);
	root->right->right->left = newNode(6);
	root->right->right->right = newNode(7);

	/*
	Constructed binary tree is:
			1
			/ \
		2 3
		/ \ \
		4 5 8
				/ \
				6 7
	*/
	
	// Function call
	cout << "Maximum width is " << getMaxWidth(root)
		<< endl;
	return 0;
}

// This code is contributed by rathbhupendra