【路飞】算法-平衡二叉树

题目链接 LeetCode leetcode-cn.com/problems/ba…

题目简介 给定一个二叉树，判断它是否是高度平衡的二叉树。
一棵高度平衡二叉树定义为：一个二叉树每个节点 的左右两个子树的高度差的绝对值不超过 1 。

题目示例 示例一

输入： root = [3,9,20,null,null,15,7]
输出： true

示例二

输入： root = [1,2,2,3,3,null,null,4,4]
输出： false

示例三

输入： root = []
输出： true

解题思路 本题要查找二叉树的每个节点左右两个子树的高度绝对值，所以：

• 采用递归的方式实现。
• 求每个子节点的高度

代码实现

/**
 * Definition for a binary tree node.
 * function TreeNode(val, left, right) {
 *     this.val = (val===undefined ? 0 : val)
 *     this.left = (left===undefined ? null : left)
 *     this.right = (right===undefined ? null : right)
 * }
 */
/**
 * @param {TreeNode} root
 * @return {boolean}
 */
 var isBalanced = function(root){
     // 根据题目要求，空节点返回true
     if(!root) return true;
     // 题目要求要小于等于1才算平衡二叉树
     // 并且若是左右子节点有一个没有，则代表不是平衡二叉树
     return Math.abs(childHeight(root.left) - childHeight(root.right)) <= 1 && isBalanced(root.left) && isBalanced(root.right)
 }
 // 求每个节点的高度
 var childHeight(root){
     if(!root) return 0; // 如果节点为null，高度为0
     // 否则就返回左右子节点的最大值+1（当前节点）
     return Math.max(childHeight(root.left), childHeight(root.right)) + 1;
 }