题目描述
输入一棵二叉树,判断该二叉树是否是平衡二叉树。
分析
平衡二叉树(Balanced Binary Tree)是二叉查找树的一个进化体,也是第一个引入平衡概念的二叉树。1962年,G.M.Adelson-Velsky 和 E.M.Landis发明了这棵树,所以它又叫AVL树。
平衡二叉树要求对于每一个节点来说,它的左右子树的高度之差不能超过1,如果插入或者删除一个节点使得高度之差大于1,就要进行节点之间的旋转,将二叉树重新维持在一个平衡状态。这个方案很好的解决了二叉查找树退化成链表的问题,把插入,查找,删除的时间复杂度最好情况和最坏情况都维持在O(logN)。但是频繁旋转会使插入和删除牺牲O(logN)左右的时间,不过相对二叉查找树来说,时间上稳定了很多。
本题中,需要从AVL树的定义入手,定义一个求树深度的函数,如果根节点root的左右子树深度之差(该差值称为平衡因子BF(Balance Factor))的绝对值小于等于1,那么继续递归,验证root的左右子树是否也符合这一点。否则,该树不是AVL。
代码
class Solution {
public:
bool IsBalanced_Solution(TreeNode* pRoot) {
if(pRoot==nullptr)
return true;
if(abs(treeDepth(pRoot->left)-treeDepth(pRoot->right))>1)
return false;
else
return IsBalanced_Solution(pRoot->left)&&IsBalanced_Solution(pRoot->right);
}
int treeDepth(TreeNode* pRoot){
if(pRoot==nullptr)
return 0;
return max(treeDepth(pRoot->left),treeDepth(pRoot->right))+1;
}
};