题目:输入一棵二叉树,判断该二叉树是否是平衡二叉树。
分析:二叉树性质:它是一棵空树或它的左右两个子树的高度差的绝对值不超过1,并且左右两个子树都是一棵平衡二叉树。从二叉树的性质入手,求出根节点的左右孩子的深度,如果差大于1就直接返回false,否则判断左右子树是否为平衡二叉树,递归重复上面的过程,求子树的深度用的是层序遍历的思想。
程序:
int deep(TreeNode *proot){
if(!proot) return 0;
vector<TreeNode *> temp;
temp.push_back(proot);
int len=1,k=1;
while(!temp.empty()){
int count=0;
while(k--){
if(temp[0]->left) {temp.push_back(temp[0]->left);count++;}
if(temp[0]->right) {temp.push_back(temp[0]->right);count++;}
temp.erase(temp.begin());
}
if(count>0) {
len++; k=count;
}
}
return len;
}
bool IsBalanced_Solution(TreeNode* pRoot) {
if(!pRoot) return true;
int l=deep(pRoot->left),r=deep(pRoot->right);
if(abs(l-r)<=1)
return IsBalanced_Solution(pRoot->left)&&IsBalanced_Solution(pRoot->right);
else return false;
}