第十五题 剑指 Offer 55 - II. 平衡二叉树

剑指 Offer 55 - II. 平衡二叉树

题目描述

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

示例1

给定二叉树 [3,9,20,null,null,15,7]

返回

true

示例2

给定二叉树 [1,2,2,3,3,null,null,4,4]

返回

false

限制

• 0 <= 树的结点个数 <= 10000

思路

• 用递归进行搜索
• 定义两个变量用于记录 左右子树的深度 左：lDep 右：rDep
• 根据平衡二叉树的定义 当左子树的深度与右子树的深度差值大于 1时 返回 false

代码（DFS）

/**
 * Definition for a binary tree node.
 * function TreeNode(val) {
 *     this.val = val;
 *     this.left = this.right = null;
 * }
 */
/**
 * @param {TreeNode} root
 * @return {boolean}
 */
var isBalanced = function(root) {
    let flag = true;
    let dfs = function(node, dep){
        if(!node) return;
        //左子树深度
        let lDep = dep;
        //右子树深度
        let rDep = dep;
        // 直至找到没有子节点为止
        if(!node.left && !node.right){
            return dep;
        }
        //该节点有左子节点 递归继续找下去 此深度+1 
        if(node.left){
           lDep = dfs(node.left, dep+1);
        }
        //该节点有右子节点 递归继续找下去 此深度+1
        if(node.right){
           rDep = dfs(node.right, dep+1);
        }
        // 左右子树深度差值大于一 不是平衡二叉树
        if(Math.abs(lDep - rDep) > 1){
            flag = false;
        }
        // 返回子树深度
        return Math.max(lDep, rDep);
    }
    dfs(root, 1);
    return flag?true:false;
};