平衡二叉树的性质:它是一棵空树或它的左右两个子树的高度差的绝对值不超过1,并且左右两个子树都是一棵平衡二叉树。
做这道题之前先说一下树的深度问题:从根结点到叶结点依次经过的结点(含根、叶结点)形成树的一条路径,最长路径的长度为树的深度。
如果我们求出树的左右子树的深度,再做差值,结果满足不大于1即说明符合平衡二叉树的性质。
如何求树的深度呢?我们可以使用递归算节点的左右子树节点取较大值,代码如下:
class Solution {
public:
int TreeDepth(TreeNode* pRoot) {
if(pRoot==nullptr) return 0;
return 1+max(TreeDepth(pRoot->left),TreeDepth(pRoot->right));
}
};
因此求解原题就可以用到求树的深度的解答,判断是否是一棵平衡二叉树,代码如下:
class Solution {
public:
bool IsBalanced_Solution(TreeNode* pRoot) {
if(pRoot==nullptr) return true;
return abs(depth(pRoot->left)-depth(pRoot->right))<=1&&
IsBalanced_Solution(pRoot->left)&&IsBalanced_Solution(pRoot->right);
}
int depth(TreeNode* pRoot)
{
if(pRoot==nullptr) return 0;
return max(depth(pRoot->left),depth(pRoot->right))+1;
}
};