平衡二叉树,又被称为AVL树(有别于AVL算法),且具有以下性质:它是一 棵空树或它的左右两个子树的高度差的绝对值不超过1,并且左右两个子树都是一棵平衡二叉树。这个方案很好的解决了二叉查找树退化成链表的问题,把插入,查找,删除的时间复杂度最好情况和最坏情况都维持在O(logN)。但是频繁旋转会使插入和删除牺牲掉O(logN)左右的时间,不过相对二叉查找树来说,时间上稳定了很多。
思想:即每个节点都要是平衡的,这棵数就是平衡二叉树。我们先定义一个Boolean型的变量来判断当前子树是否是平衡二叉树和定义一个变量来表示它当前所在层数。先依次遍历左子树,如果左子树是平衡二叉树,再依次遍历右子树。具体看代码和代码的注释。
代码:
public class Code_06_IsBalancedTree {
public static class Node {
public int value;
public Node left;
public Node right;
public Node(int data) {
this.value = data;
}
}
public static boolean isBalance(Node head) {
boolean[] res = new boolean[1];
res[0] = true;
getHeight(head, 1, res);
return res[0];
}
public static int getHeight(Node head, int level, boolean[] res) { //level表示它在第几层
if (head == null) {
return level; //如果这棵树是空树,那么它是平衡二叉树,返回它的层数
}
int lH = getHeight(head.left, level + 1, res); //使用递归判断左树是不是平衡二叉树
if (!res[0]) {
return level;
}
int rH = getHeight(head.right, level + 1, res); //判断右数是不是平衡二叉树
if (!res[0]) {
return level;
}
if (Math.abs(lH - rH) > 1) { //abs的意思是取绝对值 如果左右两树的高度差大于1就不是平衡二叉树
res[0] = false;
}
return Math.max(lH, rH); //返回这棵树的层数,它是这左右子树中大的那颗树的层数加1
}
public static void main(String[] args) {
Node head = new Node(1);
head.left = new Node(2);
head.right = new Node(3);
head.left.left = new Node(4);
head.left.right = new Node(5);
head.right.left = new Node(6);
head.right.right = new Node(7);
System.out.println(isBalance(head));
}
}
这种题目都是一个套路,要先把相对应的条件表示出来,然后使用递归依次去遍历每一个节点。可以去了解一下树形DP。