读完题目,首先想的居然是,平衡二叉树的定义是什么…
于是百度:平衡二叉树(Balanced Binary Tree)又被称为AVL树(有别于AVL算法),且具有以下性质:它是一 棵空树或它的左右两个子树的高度差的绝对值不超过1,并且左右两个子树都是一棵平衡二叉树。
接着是平衡二叉树的特性:AVL算法很好的解决了二叉查找树退化成链表的问题,把插入,查找,删除的时间复杂度最好情况和最坏情况都维持在O(logN)。但是频繁旋转会使插入和删除牺牲掉O(logN)左右的时间,不过相对二叉查找树来说,时间上稳定了很多。
其实AVL算法是针对二叉查找树的一种优化,但我好像偏离主题了。于是我读懂了平衡二叉树的定义就去做题了。
那么如何判断一棵树是不是平衡二叉树,最简单的思路就是对这个二叉树进行遍历,然后判断每个节点是否是平衡二叉树就够了呢? 于是再考虑二叉树的遍历:前序、中序、后序、层序。
剑指Offer的上一道题是计算一个二叉树的高度,猜想应该是为这道题做铺垫的吧… 于是递归了二叉树的前序遍历,只不过这次不是输出遍历节点的值,而是判断它是否是一颗平衡二叉子树。
bool IsBalanced_Solution(TreeNode* pRoot) {
if(pRoot == NULL)
return true;
if(pRoot->left != NULL)
{
IsBalanced_Solution(pRoot->left);
}
if(pRoot->right != NULL)
{
IsBalanced_Solution(pRoot->right);
}
int left = 0,right = 0;
if(pRoot->left != NULL)
{
left = getTreeheight(pRoot->left,1);
}
if(pRoot->right != NULL)
{
right = getTreeheight(pRoot->right,1);
}
if(abs(left - right) <= 1)
return true;
else return false;
}
int getTreeheight(TreeNode* pRoot,int n) //求树的高度
{
if(pRoot -> left == NULL && pRoot -> right == NULL)
{
return n;
}
int a = 0,b = 0;
if(pRoot->left != NULL)
{
a = getTreeheight(pRoot->left,n+1);
}
if(pRoot->right != NULL)
{
b = getTreeheight(pRoot->right,n+1);
}
return max(a,b);
}