leetcode 110. 平衡二叉树
给定一个二叉树,判断它是否是高度平衡的二叉树。
本题中,一棵高度平衡二叉树定义为:一个二叉树每个节点 的左右两个子树的高度差的绝对值不超过 1 。
输入:root = [3,9,20,null,null,15,7]
输出:true
输入:root = [1,2,2,3,3,null,null,4,4]
输出:false
递归
参数和返回值:树的一个节点 TreeNode*, 返回当前树的深度
递归的单层逻辑:当前节点的左右子树高度差不大于1,返回当前节点为根的高度,否则返回 -1,说明不是平衡二叉树
终止条件:当前节点为空
int getDepth(TreeNode* node){
if(node == nullptr)
return 0;
int leftDepth = getDepth(node->left);
if(leftDepth == -1)
return -1;
int rightDepth = getDepth(node->right);
if(rightDepth == -1)
return -1;
if(abs(leftDepth - rightDepth) > 1) //左右子树深度差大于1
return -1;
return 1 + max(leftDepth, rightDepth);
}
bool isBalanced(TreeNode* root) {
return getDepth(root) != -1;
}