【二叉树】03 判断是否是平衡二叉树+求树的深度

判断是否是平衡二叉树

判断是否是平衡二叉树，就是判断二叉树左右子树深度之差不大于1.

int level(TreeNode* root) {
	if (root == NULL)
		return 0;
	return max(level(root->left), level(root->right)) + 1;
}
bool isBalanced(TreeNode* root) {
	if (root == NULL) {
		return true;
	}
	int tmp = abs(level(root->left)-level(root->right));
	return tmp < 2 && isBalanced(root->left) && isBalanced(root->right);
}

优化方式

一种改进的方式是，可以考虑利用动态规划的思想，将TreeNode指针作为key，高度作为value，
一旦发现节点已经被计算过，直接返回结果，这样，level函数对每个节点只计算一次。

另一种更为巧妙的方法是，isBlaned返回当前节点的高度，用-1表示树不平衡。
将计算结果自底向上的传递，并且确保每个节点只被计算一次，复杂度为O（n）；

int isBalancedHelper(TreeNode* root) {
	if (root == NULL) {
		return 0;
	}
	int leftHeight = isBalancedHelper(root->left);
	if (leftHeight == -1) {
		return -1;
	}
	int rightHeight = isBalancedHelper(root->right);
	if (rightHeight == -1) {
		return -1;
	}
	if (abs(leftHeight - rightHeight) > 1) {
		return -1;
	}
	return max(leftHeight , rightHeight) + 1;//返回的高度
}
bool isBalancedTree(TreeNode* root) {
	return (isBalancedHelper(root) != -1);
}

计算树的深度

int treeDepth(TreeNode* node) {
	if (node == NULL) {
		return 0;
	}
	else {
		return max(treeDepth(node->left), treeDepth(node->right)) + 1;
	}
}