平衡二叉树
输入一棵二叉树的根节点,判断该树是不是平衡二叉树。如果某二叉树中任意节点的左右子树的深度相差不超过1,那么它就是一棵平衡二叉树。
示例 1:
给定二叉树 [3,9,20,null,null,15,7]
3
/ \
9 20
/ \
15 7
返回 true 。
示例 2:
给定二叉树 [1,2,2,3,3,null,null,4,4]
1
/ \
2 2
/ \
3 3
/ \
4 4
返回 false 。
限制:
- 1 <= 树的结点个数 <= 10000
解法: 平衡树的定义是,每个结点的左右子树的深度差不超过1.一种是从根向叶结点走,一种是从叶结点往根节点走。
Solution 1:
利用上面一题的求解树深度的方法,从根节点往叶子结点走,进行递归判断。
但是会存在重复计算的弊端,导致计算量较大。
class Solution {
public:
bool isBalanced(TreeNode* root) {
if(root == nullptr)
return true;
int left = maxDepth(root->left);
int right = maxDepth(root->right);
if(abs(left-right) > 1)
return false;
return isBalanced(root->left) && isBalanced(root->right);
}
int maxDepth(TreeNode* root) {
if(root == nullptr)
return 0;
int leftDepth = maxDepth(root->left);
int rightDepth = maxDepth(root->right);
return max(leftDepth,rightDepth) + 1;
}
};
根据后续遍历从下向上计算进行判断,从叶子结点往根结点方向走,可以减少重复的计算。
class Solution {
public:
bool isBalanced(TreeNode* root) {
if (root == nullptr)
return true;
int depth = 0;
return isBalancedCore(root, depth);
}
bool isBalancedCore(TreeNode* root,int& depth){
if(root == nullptr)
return true;
int left = 0;
int right = 0;
int diff;
if(isBalancedCore(root->left,left) && isBalancedCore(root->right,right)){
diff = abs(left - right);
if(diff <= 1){
depth = max(left,right) + 1; // 更新结点的深度信息
return true;
}
}
return false;
}
};