本题思路是针对高度,只不过是判断高度之差的绝对值是否大于1罢了,这里引入特殊变量-1用来说明不是平衡二叉树
如果左右子节点对应的子树有一个不是平衡二叉树就一直-1返回到头了。
只有全是平衡二叉树时才正常运行完。
class Solution {
public:
int traversal(TreeNode* cur){
if(cur == nullptr) return 0;
int l_dept = traversal(cur->left);
if(l_dept == -1) return -1;
int r_dept = traversal(cur->right);
if(r_dept == -1) return -1;
return abs(r_dept-l_dept)>1? -1:1+max(r_dept,l_dept);
}
bool isBalanced(TreeNode* root) {
return traversal(root) != -1;
}
};