题目描述
输入一棵二叉树,判断该二叉树是否是平衡二叉树。
思路:用一个全局变量记录遍历到的节点的当前深度。使用后序遍历的方式,当遍历到当前节点时,我们已经遍历了它的左右子数,所以只需要在遍历每个节点的时候记录下他们的深度,我们就可以根据节点左右字数的深度差来判断树是否平衡。
public class Solution {
public boolean IsBalanced_Solution(TreeNode root) {
int depth = 0;
return isBalance(root, 0);
}
public boolean isBalance(TreeNode root, int depth) {
if (root == null)
return true;
int left = 0;//先给深度赋初值,然后在后续遍历递归的时候会更新深度值
int right = 0;
if (isBalance(root.left, left) && isBalance(root.right, right)){
int diff = left - right;
if (diff <= 1 || diff >= -1){
depth = left > right ? left + 1 : right + 1;//记录深度
return true;
}
}
return false;
}
}