剑指offer：面试题55-2：平衡二叉树

题目

输入一棵二叉树的根结点，判断该树是不是平衡二叉树。如果某二叉树中任意结点的左右子树的深度相差不超过1，那么它就是一棵平衡二叉树。

思路

首先，明确平衡二叉树的定义：它是一棵空树或它的左右两个子树的高度差的绝对值不超过1，并且左右两个子树都是一棵平衡二叉树。

• 思路一，在55-1：树的深度的基础上，遍历每个节点，并判断每个节点的左右子树高度差是否超过1，若超过，则不是平衡二叉树；若所有节点都不超过，则是平衡二叉树。
• 思路二，上述思路显然一个节点会被遍历多次。因此思路二给出只遍历一次的思路（且有剪枝）。即使用后序遍历的方式遍历二叉树的节点，这样遍历父节点时已经遍历过其左右子树，只需要记录下每个节点的深度，即可在一次后序遍历后，知道这棵树是否平衡。

测试用例

1.功能测试（左斜树、右斜树、平衡或者不平衡树）
3.特殊测试（一个结点，null）

Java实现

实现一

public class IsBalancedTree {
    public static class TreeNode{
        int val = 0;
        TreeNode left = null;
        TreeNode right = null;

        public TreeNode(int val) {
            this.val = val;
        }
    }

    /**
     * 方法一
     * @param root
     * @return
     */
    public boolean IsBalanced_Solution(TreeNode root) {
        if (root == null) return true;
        if (Math.abs(treeDepth(root.right) - treeDepth(root.left)) <= 1) {
            return IsBalanced_Solution(root.left) && IsBalanced_Solution(root.right);
        } else {
            return false;
        }
    }


    public int treeDepth(TreeNode root) {
        if (root == null) {
            return 0;
        }
        int leftDepth = treeDepth(root.left);
        int rightDepth = treeDepth(root.right);
        return 1 + Math.max(leftDepth, rightDepth);
    }
}

实现二

public class IsBalancedTree {
    public static class TreeNode{
        int val = 0;
        TreeNode left = null;
        TreeNode right = null;

        public TreeNode(int val) {
            this.val = val;
        }
    }
    /**
     * 方法二
     * 利用后续遍历，能够先遍历子节点，记录下深度，再计算父节点是否balance，就能够只遍历一遍。
     * @param root
     * @return
     */
    public boolean IsBalanced_Solution(TreeNode root) {
        if (root == null) return true;
        return treeDepthPruning(root) != -1;
    }


    public int treeDepthPruning(TreeNode root) {
        if (root == null) {
            return 0;
        }
        int leftDepth = treeDepth(root.left);
        if (leftDepth == -1) return -1;          // 这是剪枝
        int rightDepth = treeDepth(root.right);
        if (rightDepth == -1) return -1;         // 这是剪枝
        if (Math.abs(leftDepth - rightDepth) > 1) {
            return -1;
        } else {
            return 1 + Math.max(leftDepth, rightDepth);
        }
    }

}

收获

1. 判断树不平衡后，可以直接退出（剪枝），不需要遍历完整棵树。