给定一个二叉树,判断它是否是高度平衡的二叉树。
本题中,一棵高度平衡二叉树定义为:
一个二叉树每个节点 的左右两个子树的高度差的绝对值不超过1。
版本1
sol函数来计算给定节点的高度,然后在isBalanced函数中递归调用,判断每个节点的子节点之间的高度差是否大于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;
if(abs(sol(root->left)-sol(root->right))>1)
return false;
else
{
bool ret1=1,ret2=1;
if(root->left)
ret1=isBalanced(root->left);
if(root->right)
ret2=isBalanced(root->right);
if(!ret1||!ret2)
return false;
else
return true;
}
}
int sol(TreeNode* root)
{
if(root==NULL)
return 0;
return max(sol(root->left),sol(root->right))+1;
}
};
优化版本1
class Solution {
public:
int height(TreeNode* root) {
if (root == NULL) {
return 0;
} else {
return max(height(root->left), height(root->right)) + 1;
}
}
bool isBalanced(TreeNode* root) {
if (root == NULL) {
return true;
} else {
return abs(height(root->left) - height(root->right)) <= 1 && isBalanced(root->left) && isBalanced(root->right);
}
}
};
最优版本(从底至顶)
从左边最底节点判断是否为平衡树,是的话返回节点高度,否的话返回-1,
height(root)<0说明有节点不满足条件
/**
* 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) {
return height(root)>=0;
}
int height(TreeNode* root)
{
if(root==NULL)
return 0;
int h1=height(root->left);
if(h1==-1)//左子树不满足直接返回,省去计算右子树时间
return -1;
int h2=height(root->right);
if(h2==-1||abs(h1-h2)>1)
return -1;
else
return max(h1,h2)+1;
}
};