题目
输入一棵二叉树,判断该二叉树是否是平衡二叉树。
先看看平衡二叉树的定义:
(1) 任一结点的左、右子树均为AVL树
(2) 根结点左、右子树高度差绝对值不超过1
思路 判断每个节点的左右子树的高度差,若高度差绝对值大于1,则不是平衡二叉树
public class Solution {
public boolean IsBalanced_Solution(TreeNode root) {
if(root==null) return true;
int left=getmax(root.left);
int right=getmax(root.right);
if(Math.abs(left-right)>1&&IsBalanced_Solution(root.left)&&IsBalanced_Solution(root.right))
return false;
return true;
}
public int getmax(TreeNode root){
if(root==null) return 0;
int left=getmax(root.left);
int right=getmax(root.right);
return Math.max(left,right)+1;
}
}
这个算法有个明显的缺点,就是从头节点开始,从上往下遍历,每次遍历都重复计算的下层的节点。
因此可以改为从下往上遍历节点,若最下层节点不是AVL树,则上层也不是AVL树,返回false,这样可以避免大量重复计算
public class Solution {
public boolean IsBalanced_Solution(TreeNode root) {
if(getmax(root)==-1)
return false;
return true;
}
public int getmax(TreeNode root){
if(root==null) return 0;
int left=getmax(root.left);
if(left==-1) return -1;
int right=getmax(root.right);
if(right==-1) return -1;
return Math.abs(left-right)>1? -1:1+Math.max(left,right);
}
}
其他的题题解
剑指offer题解 java版