T110 平衡二叉树

Posted 2021-02-13 rainbow-

给定一个二叉树，判断它是否是高度平衡的二叉树。
本题中，一棵高度平衡二叉树定义为：
一个二叉树每个节点 的左右两个子树的高度差的绝对值不超过1。
示例 1:
给定二叉树 [3,9,20,null,null,15,7]
    3
   / 
  9  20
    /  
   15   7
返回 true 。

示例 2:
给定二叉树 [1,2,2,3,3,null,null,4,4]
       1
      / 
     2   2
    / 
   3   3
  / 
 4   4
返回 false 。

思路 : 递归

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

 1 class Solution110 {
 2 
 3   public boolean isBalanced(TreeNode root) {
 4     return search(root, 1) != null;
 5   }
 6 
 7   Integer search(TreeNode root, int depth) {
 8     if (root != null) {
 9       Integer leftDepth = search(root.left, depth + 1);
10       Integer rightDepth = search(root.right, depth + 1);
11       /*
12       如果不为平衡二叉树,返回null
13       如果此树为平衡二叉树: 左子树为平衡二叉树&&右子树为平衡二叉树&&左子树和右子树高度差<=1,返回左子树和右子树的高度大的那个+1
14        */
15       return (leftDepth != null && rightDepth != null && Math.abs(leftDepth - rightDepth) <= 1) ?
16         Math.max(leftDepth, rightDepth) + 1 : null;
17     }
18     //如果此树为空树,它仍是一颗平衡二叉树,高度为0
19     return 0;
20   }
21 }