题目
输入一棵二叉树,判断该二叉树是否是平衡二叉树。
在这里,我们只需要考虑其平衡性,不需要考虑其是不是排序二叉树
思路
从下往上遍历,如果子树是平衡二叉树,则返回子树的高度;如果发现子树不是平衡二叉树,则直接停止遍历,这样至多只对每个结点访问一次。
public class Solution {
public boolean IsBalanced_Solution(TreeNode root) {
return !(-1==getDepth(root));
}
private 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(right-left)>1)? -1:Math.max(left,right)+1;
}
}
未剪枝,每个点进行遍历
public classSolution {
public boolean IsBalanced_Solution(TreeNode root) {
if(root == null) {
return true;
}
return Math.abs(maxDepth(root.left) - maxDepth(root.right)) <= 1 &&
IsBalanced_Solution(root.left) && IsBalanced_Solution(root.right);
}
private int maxDepth(TreeNode root) {
if(root == null) {
return 0;
}
return 1 + Math.max(maxDepth(root.left), maxDepth(root.right));
}
}