题目:实现一个函数,检查二叉树是否平衡,平衡的定义如下,对于树中的任意一个结点,其两颗子树的高度差不超过1。给定指向树根结点的指针TreeNode* root,请返回一个bool,代表这棵树是否平衡。
思路:根据题目定义,子树的高度差超过1即为不平衡,所以可以写一个求子树高度的函数,用递归实现比较方便
/*
struct TreeNode {
int val;
struct TreeNode *left;
struct TreeNode *right;
TreeNode(int x) :
val(x), left(NULL), right(NULL) {
}
};*/
class Balance {
public:
int getHeight(TreeNode* node){
if(node==NULL)
return 0;
else
return getHeight(node->left)>getHeight(node->right)?getHeight(node->left)+1:getHeight(node->right)+1;
}
bool isBalance(TreeNode* root) {
// write code here
if(root==NULL)
return true;
int heightDiff = getHeight(root->left)-getHeight(root->right);
if(heightDiff<-1 || heightDiff>1)
return false;
else
return (isBalance(root->left) && isBalance(root->right));
}
};