"Coding Interview Guide" -- 判断二叉树是否为平衡二叉树

 1 public boolean isBalanceTree(Node head)
 2 {
 3     if(head == null)             // 平衡二叉树可以是空树
 4     {
 5     return true;
 6     }
 7 
 8     boolean[] res = new boolean[1];      // res记录是否为平衡二叉树
 9     res[0] = true;
10     getHeight(head, 1, res);
11     return res[0];
12 }
13 
14 public int getHeight(Node head, int level, boolean[] res)
15 {
16     if(head == null)
17     {
18     return level;
19     }
20 
21     int l = getHeight(head.left, level+1, res);
22     if(!res[0])             // 只要任一子树不是平衡二叉树，则整个二叉树就不是平衡二叉树，直接返回
23     {
24     return level;     
25     }
26     int r = getHeight(head.right, level+1, res);
27     if(!res[0])
28     {
29     return level;
30     }
31     if(Math.abs(l - r) > 1)         // 高度差的绝对值超过1，不是平衡二叉树
32     {
33     res[0] = false;
34     }
35 
36     return Math.max(l, r);
37 }