开始想这道题的时候,想着使用bfs,然后遍历层数,通过每层节点数量和预期的数量的差距来找。按道理应该可以实现,但是实现的逻辑难度太大,遂放弃。
然后看答案,看完发现是一个典型的DFS,只不过有点变种,首先找到叶子节点,然后当前返回0,然后最后通过判断如果left-right的绝对值大于1直接返回-1,如果小于等于1,那么就在结果上加1
class Solution {
public boolean isBalanced(TreeNode root) {
return find(root)!=-1;
}
private int find(TreeNode root){
if(root==null) return 0;
int left = find(root.left);
if(left==-1) return -1;
int right = find(root.right);
if(right==-1) return -1;
return Math.abs(left-right)>1?-1:Math.max(left,right)+1;
}
}