文章目录
题目描述
输入一棵二叉树的根节点,判断该树是不是平衡二叉树。如果某二叉树中任意节点的左右子树的深度相差不超过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则返回false
代码实现:
class Solution {
public:
bool isBalanced(TreeNode* root) {
if(!root) return true;
return (abs(maxDepth(root->left) - maxDepth(root->right)) <= 1) && isBalanced(root->left) && isBalanced(root->right);
}
private:
int maxDepth(TreeNode* root){
if(!root) return 0;
return max(maxDepth(root->left), maxDepth(root->right)) + 1;
}
};
方法二:后序遍历
- 使用后序遍历的方式遍历二叉树的每个节点,那么在遍历到一个节点之前我们就已经
- 遍历了它的左右子树。只要在每个节点的时候记录它的深度(某一节点的深度等于它到叶节点的路径的长度)
- 就可以判断每个节点是不是平衡的
代码实现:
class Solution {
public:
bool isBalanced(TreeNode* root) {
if(!root) return true;
int depth = 0;
return isBalanced(root, depth);
}
private:
bool isBalanced(TreeNode* root, int& depth){
if(root == nullptr){
depth = 0;
return true;
}
int l = 0, r = 0;
if(isBalanced(root->left, l) && isBalanced(root->right, r)){
int dif = abs(l - r);
if(dif <= 1){
depth = (l > r ? l : r) + 1;
return true;
}
}
return false;
}
};
如有帮助到您,可以多多点赞、评论鼓励哟~~~