LeetCode刷题之路-动态规划-剑指 Offer 55 - II. 平衡二叉树

前言

又是一天美丽的意外，反正二叉树也用的少，没怎么接触。

题目描述

---

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

---

解题思路

刚开始，第一反应还是用地龟。提交的时候，棋差一着啊。。只判断了顶级节点的树的深度差，没有处理左右子节点的深度差。但是一想，感觉这复杂度也太高了点吧。相当于每个节点都要地龟计算深度，然后再每个节点判断是否为完全二叉树。然后也没其他思路，所以就看了下题解。
真的妙啊。官方术语叫，地龟 + 剪枝
我理解下来，就是在计算深度的同时，通过一个特殊值 -1 来表示该节点不是平衡二叉树。从而加快运算。

解题代码

public boolean isBalanced(TreeNode root) {
        // 在APP的首页看到这题了，还以为是每日一题...
        // 明显地龟走起
        if (root == null) {
            return true;
        }
        
        // 地龟 + 剪枝法
        // 秘密在于：在计算深度的同时，用一个特殊值：-1，代表不是平衡子枝
        return depth(root) != -1;
    }

    public int depth(TreeNode parent) {
        if (parent == null) {
            return 0;
        }

        int left = depth(parent.left);
        if (left == -1) {
            // 如果left的子枝不是平衡二叉树，那就肯定不是平衡二叉树
            return -1;
        }
        int right = depth(parent.right);
        if (right == -1) {
            // 如果right的子枝不是平衡二叉树，那就肯定不是平衡二叉树
            return -1;
        }

        // 判断当前节点是否为平衡二叉树
        // 不是，则返回-1
        // 是，则计算当前二叉树的深度：left和right的最大值，+1（因为当前节点存在，所以至少为1）
        return Math.abs(left - right) > 1 ? -1 : Math.max(left, right) + 1;
    }