判断是否是平衡二叉树
判断是否是平衡二叉树,就是判断二叉树左右子树深度之差不大于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;
}
}