给定一个二叉树,判断它是否是高度平衡的二叉树。
本题中,一棵高度平衡二叉树定义为:
一个二叉树每个节点 的左右两个子树的高度差的绝对值不超过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
。
题目大意:判断一棵二叉树是否是平衡的。
题目分析:通过dfs先序遍历二叉树的节点,同时记录节点的高度,一旦发现一个节点的左右子树高度差超过1,就可以判断该二叉树是非平衡的。
代码展示:
class Solution {
public:
bool isBalanced(TreeNode* root) {
if(root==NULL)
return true;
bool flag = true;
dfs(root,flag);
return flag;
}
int dfs(TreeNode* root,bool& flag){
if(flag==false)
return -1;
if(root==NULL)
return 0;
int left = dfs(root->left,flag);
int right = dfs(root->right,flag);
if(abs(left-right)>1)
flag = false;
return max(left,right)+1;
}
};