算法 - 剑指Offer II. 平衡二叉树

题目

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

解题思路

这题第一反应是使用之前树的深度的递归， 去计算左子树深度减去右子树深度如果大于1， 则不平衡， 直接去计算其深度的递归函数， 再去递归其本身的左右子树即可， 这里的解题函数为isBalancedSample， 还有一种解题就是用常规的递归去做， 我们去判断他的返回值， 如果大于0， 则是树的深度为平衡二叉树， 如果返回小于0， 则不是平衡二叉树， 首先添加判断， 遍历到节点为空都为平衡节点，然后去计算其左右节点各个的深度， 这里计算深度也和之前将的一样使用1+max(depth(root.left), depth(root.right))类似， 然后如果左右节点深度相减的绝对值为大于1的， 则直接返回负数， 判断为不平衡节点， 代码如下。

Java代码实现

public class IsBalanced {
    public class TreeNode {
       int val;
       TreeNode left;
       TreeNode right;
       TreeNode(int x) { val = x; }
   }
    public boolean isBalancedSample(TreeNode root) {
        if(root == null){
            return true;
        }
        return Math.abs(depthDFS(root.left) - depthDFS(root.right)) <= 1 && isBalanced(root.left) && isBalanced(root.right) ;
    }
    public int depthDFS(TreeNode root) {
        if(root == null){
            return 0;
        }
        return Math.max(depthDFS(root.left), depthDFS(root.right)) + 1;
    }
    public boolean isBalanced(TreeNode root) {
        return depth(root) >= 0;
    }
    public int depth(TreeNode root) {
        if(root == null){
            return 0;
        }
        int left = depth(root.left);
        int right = depth(root.right);
        if(left == -1 || right == -1 || Math.abs(left - right) > 1){
            return -1;
        }
        return 1 + Math.max( left, right);
    }
}