55.-II 平衡二叉樹
面試題55 - II. 平衡二叉樹
難度簡單19
輸入一棵二叉樹的根節點,判斷該樹是不是平衡二叉樹。如果某二叉樹中任意節點的左右子樹的深度相差不超過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 。
private boolean isBalanced = true;//默認爲true
public boolean isBalanced(TreeNode root) {
dfs(root);
return isBalanced;
}
private int dfs(TreeNode root){
if(root == null || !isBalanced){
return 0;
}
int left = dfs(root.left); //計算左子節點長度
int right = dfs(root.right);//計算右子節點長度
if(Math.abs(left-right)>1){//如果左右子節點長度差大於1 則爲false
isBalanced = false;
}
return 1+Math.max(left,right);
}