本文参考自《剑指offer》一书,代码采用Java语言。
题目
输入一棵二叉树的根结点,判断该树是不是平衡二叉树。如果某二叉树中任意结点的左右子树的深度相差不超过1,那么它就是一棵平衡二叉树。
思路
在(55-1) 二叉树的深度基础上修改:计算树的深度,树的深度=max(左子树深度,右子树深度)+1。在遍历过程中,判断左右子树深度相差是否超过1,如果不平衡,则令树的深度=-1,用来表示树不平衡。最终根据树的深度是否等于-1来确定是否为平衡树。
测试算例
1.功能测试(左斜树、右斜树、平衡或者不平衡树)
3.特殊测试(一个结点,null)
Java代码
//题目:输入一棵二叉树的根结点,判断该树是不是平衡二叉树。如果某二叉树中
//任意结点的左右子树的深度相差不超过1,那么它就是一棵平衡二叉树。
public class BalancedBinaryTree {
public class TreeNode {
int val = 0;
TreeNode left = null;
TreeNode right = null;
public TreeNode(int val) {
this.val = val;
}
}
public boolean IsBalanced_Solution(TreeNode root) {
return getDepth(root)!=-1;
}
public int getDepth(TreeNode root) {
if(root==null) return 0;
int left=getDepth(root.left);
if(left==-1) return -1;
int right=getDepth(root.right);
if(right==-1) return -1;
return Math.abs(left - right) > 1 ? -1 : 1 + Math.max(left, right);
}
/*
//自己开始想的方法,但是一定要把树给遍历完才行;上面的方法实现了剪枝
boolean isBalanced=true;
public boolean IsBalanced_Solution(TreeNode root) {
TreeDepth(root);
return isBalanced;
}
public int TreeDepth(TreeNode root) {
if(root==null)
return 0;
int left=TreeDepth(root.left);
int right=TreeDepth(root.right);
if(left-right>1 || right-left>1)
isBalanced=false;
return Math.max(left+1,right+1);
}
*/
}
收获
1.在判断出树不平衡后,进行剪枝(即代码中直接返回-1,不再对其他子树进行判断),以提高效率。