有一棵二叉树,请设计一个算法判断这棵二叉树是否为平衡二叉树。
给定二叉树的根结点root,请返回一个bool值,代表这棵树是否为平衡二叉树。
/*
struct TreeNode {
int val;
struct TreeNode *left;
struct TreeNode *right;
TreeNode(int x) :
val(x), left(NULL), right(NULL) {
}
};*/
class CheckBalance {
public:
int getHeight(TreeNode* root)
{
if(root==NULL)
return 0;
int left=getHeight(root->left);
int right=getHeight(root->right);
return 1+max(left,right);
}
bool check(TreeNode* root) {
// write code here
if(root==NULL)
return true;
if(!check(root->left))
return false;
if(!check(root->right))
return false;
int left=getHeight(root->left);
int right=getHeight(root->right);
if(abs(left-right)<=1)
return true;
else
return false;
}
};