leetcode110. 平衡二叉树
给定一个二叉树,判断它是否是 平衡二叉树
示例 1:
输入:root = [3,9,20,null,null,15,7]
输出:true
示例 2:
输入:root = [1,2,2,3,3,null,null,4,4]
输出:false
示例 3:
输入:root = []
输出:true
算法思路
分为两个函数,一个函数递归计算二叉树的高度,一个函数递归判断二叉树是否平衡。
isBalanced 函数
功能:判断二叉树是否平衡。
逻辑:使用递归方法判断二叉树是否平衡。
如果 root 是 NULL,返回 true,因为空树是平衡的。
否则,首先计算左子树和右子树的高度差,如果高度差大于1,则树不平衡,返回 false。
接着,递归地判断左子树 root->left 和右子树 root->right 是否平衡。
如果左子树和右子树都平衡,且它们的高度差不超过1,则当前树也是平衡的,返回 true。
height 函数
功能:计算二叉树的高度。
逻辑:使用递归方法计算二叉树的高度。
如果 root 是 NULL,说明是空树,返回高度为0。
否则,递归计算左子树的高度 height(root->left) 和右子树的高度 height(root->right)。
返回左子树和右子树中较大者的高度加1,作为当前节点的高度。
算法步骤
计算高度:
从根节点开始,递归地计算每个子树的高度。
对于每个节点,其高度是其左子树和右子树高度的较大者加1。
判断平衡性:
从根节点开始,检查其左子树和右子树的高度差。
如果高度差大于1,立即返回 false,因为这意味着树不平衡。
否则,继续递归地检查左子树和右子树的平衡性。
如果左子树和右子树都返回 true,则当前节点的树也是平衡的。
具体代码
class Solution {
public:
int height(TreeNode* root) {
if (root == NULL) {
return 0;
} else {
return max(height(root->left), height(root->right)) + 1;
}
}
bool isBalanced(TreeNode* root) {
if (root == NULL) {
return true;
} else {
return abs(height(root->left) - height(root->right)) <= 1 &&
isBalanced(root->left) && isBalanced(root->right);
}
}
};