题目
July 7, 2021 · View on GitHub
输入一棵二叉树的根结点,判断该树是不是平衡二叉树。
如果某二叉树中任意结点的左右子树的深度相差不超过 1,那么它就是一棵平衡二叉树。
注意:
规定空树也是一棵平衡二叉树。
样例
输入:二叉树[5,7,11,null,null,12,9,null,null,null,null]如下所示,
5
/ \
7 11
/ \
12 9
输出:true
参考答案
``c++ class Solution { public: bool isBalanced(TreeNode* root) { /* unit test root is nil root not nil, left is nil, right is nil root not nil, left not nil, right nil. */ int height=dfs(root); if(height>=0) return true; else return false; }
// 当非平衡时,return -1; 平衡时,return high;
// 首先判断左子树平衡与否,再判断右子树平衡与否,在判断整棵树平衡与否;
int dfs(TreeNode *root){
if(!root) return 0;
int left=dfs(root->left);
if(left<0) return -1;
int right=dfs(root->right);
if(right<0) return -1;
if(abs(left-right)>1) return -1;
return max(left,right)+1;
}
};