Balance Tree

private int maxChildHeight(TreeNode<T> node) {
	if(node != null) {
		return 1 + Math.max(maxChildHeight(node.getLeft()), maxChildHeight(node.getRight()));
	}
	return 0;
}

private int getLeftHeight() {
	return maxChildHeight(this.getLeft());
}

private int getRightHeight() {
	return maxChildHeight(this.getRight());
}

private int getBalanceFactor() {
	return  getRightHeight() - getLeftHeight();
}

private TreeState getState() {
	if(this.getLeftHeight() - this.getRightHeight() > 1) {
		return TreeState.LeftHeavy;
	}
	if(this.getRightHeigth() - this.getLeftHeight() > 1) {
		return TreeState.RightHeavy;
	}

	return TreeState.Balanced;
}

private void balance() {
	if(this.getState() == TreeState.RightHeavy) {
		if(this.getRight() != null && this.getRight().getBalanceFactor < 0) {
			leftRightRotation();
		} else {
			leftRotation();
		}
	}
    if(this.getState() == TreeState.LeftHeavy) {
		if(this.getLeft() != null && this.getLeft().getBalanceFactor < 0) {
			rightLeftRotation();
		} else {
			RightRotation();
		}
	}
}