题目:
Given a binary tree, determine if it is height-balanced.
For this problem, a height-balanced binary tree is defined as a binary tree in which the depth of the two subtrees of every node never differ by more than 1.
题意:
给定一棵二叉树,判断是否是高度平衡的。
高度平衡的意思是,对于每一个节点,左子树跟右子树的高度最多相差1。
思路:
- 先写一个求高度的函数,递归的求该节点的高度,height(root) = 1 + max(height(root->left),height(root->right))。然后递归判断是否高度平衡,如果当前节点的左子树右子树高度相差小于1,那么递归判断左孩子与右孩子是否满足高度平衡。
以上。
代码如下:
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
class Solution {
public:
bool isBalanced(TreeNode* root) {
if(root == NULL)return true;
int diff = depth(root->left) - depth(root->right);
if( diff >= -1 && diff <= 1)return isBalanced(root->left)&&isBalanced(root->right);
else return false;
}
int depth(TreeNode* root){
if(root == NULL)return 0;
else return max(depth(root->left),depth(root->right)) + 1;
}
};
- 以上代码会对同一个节点多次去求高度,算法的复杂度是需要遍历n个节点,并且求每个节点高度时需要平均探索lg(n)的高度,所以复杂度是O(nlgn)。所以我们应该对每个节点只求一次高度,很明显递归时是从下面的节点往上,上面的节点的高度是依靠下面的节点的高度的结果。在求根节点的高度的时候,会去递归求左右孩子的高度,然后取其中一个较大的值再加上1。其实在求左右孩子的高度的时候,可以判断左右子树是否平衡,如果不平衡的话返回高度值是-1代表子树不平衡,所以不是平衡的,那么递归会立即停止执行并向上继续返回-1。如果左右子树都平衡,再去查看左右子树的差值。这样子就不需要重复计算节点的高度。
以上。
代码如下:
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
class Solution {
public:
bool isBalanced(TreeNode* root) {
if(root == NULL)return true;
return (checkHeight(root) > 0);
}
int checkHeight(TreeNode* root) {
if(root == NULL)return 0;
int left = checkHeight(root->left);
if(left == -1)return -1;
int right = checkHeight(root->right);
if(right == -1)return -1;
int diff = left - right;
if(diff > 1 || diff < -1)return -1;
return max(left, right) + 1;
}
};