题目描述

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

示例1:
No.110 平衡二叉树 - 图1

  1. 输入:root = [3,9,20,null,null,15,7]
  2. 输出:true

示例2:
No.110 平衡二叉树 - 图2

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

示例3:

  1. 输入:root = []
  2. 输出:true

思路

和求最大深度一致,如果左子树的最大深度和又子树的最大深度之差 >= 2,那么就不平衡

  1. /**
  2. * Definition for a binary tree node.
  3. * public class TreeNode {
  4. * int val;
  5. * TreeNode left;
  6. * TreeNode right;
  7. * TreeNode() {}
  8. * TreeNode(int val) { this.val = val; }
  9. * TreeNode(int val, TreeNode left, TreeNode right) {
  10. * this.val = val;
  11. * this.left = left;
  12. * this.right = right;
  13. * }
  14. * }
  15. */
  16. class Solution {
  17. boolean result = true;
  18. public boolean isBalanced(TreeNode root) {
  19. dfs(root);
  20. return result;
  21. }
  22. private int dfs(TreeNode root) {
  23. if (root == null) {
  24. return 0;
  25. }
  26. int leftMax = dfs(root.left);
  27. int rightMax = dfs(root.right);
  28. if (Math.abs(leftMax - rightMax) >= 2) {
  29. result = false;
  30. }
  31. return Math.max(leftMax, rightMax) + 1;
  32. }
  33. }

注意事项:

  1. 求绝对值: Math.abs(int, int)