题目
给定一个二叉树,判断它是否是高度平衡的二叉树。
本题中,一棵高度平衡二叉树定义为:
一个二叉树每个节点 的左右两个子树的高度差的绝对值不超过 1 。
示例 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
提示:
- 树中的节点数在范围 [0, 5000] 内
- − 1 0 4 < = N o d e . v a l < = 1 0 4 -10^4 <= Node.val <= 10^4 −104<=Node.val<=104
分析
此题理解了二叉树平衡的定义即可。根据定义是左右子树的高度差是-1,0或者1。因此此题转换为计算二叉树的高度。
对于二叉树常用递归方法来计算树的高度。
实现
C语言代码实现如下
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* struct TreeNode *left;
* struct TreeNode *right;
* };
*/
static inline int MaxFunc(int a, int b)
{
return (a > b) ? a : b;
}
int GetTreeHeight(struct TreeNode* root)
{
if (root == NULL) {
return 0;
}
return 1 + MaxFunc(GetTreeHeight(root->left), GetTreeHeight(root->right));
}
bool isBalanced(struct TreeNode* root){
int lheight;
int rheight;
if (root == NULL) {
return true;
}
lheight = GetTreeHeight(root->left);
rheight = GetTreeHeight(root->right);
return abs(lheight - rheight) <= 1 && isBalanced(root->left) && isBalanced(root->right);
}