剑指offer55-Ⅱ 平衡二叉树【DFS】

题目描述

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

示例 1:

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

1
2
3
4
5
  3
/ \
9 20
/ \
15 7

返回 true

示例 2:

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

1
2
3
4
5
6
7
      1
/ \
2 2
/ \
3 3
/ \
4 4

返回 false

限制:

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

解题思路

方法一:深度优先搜索

dfs方法,后序遍历,计算出左右子树的长度,判断左右子树的高度差的绝对值是否小于1,即判断是否为平衡二叉树,递归。

代码实现

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
// -1为不平衡状态,平衡状态下返回树的高度
public boolean isBalanced(TreeNode root) {
return maxDepth(root) != -1;
}

private int maxDepth(TreeNode root) {
if (root == null)
return 0;
int left = maxDepth(root.left);
if(left == -1)
return -1;
int right = maxDepth(root.right);
if(right == -1)
return -1;
return Math.abs(left - right) < 2 ? Math.max(left, right) + 1 : -1;
}

复杂度分析

时间复杂度:O(n)

空间复杂度:O(n) 递归栈

资料

Author

John Doe

Posted on

2021-05-31

Updated on

2021-06-08

Licensed under

You need to set install_url to use ShareThis. Please set it in _config.yml.
You forgot to set the business or currency_code for Paypal. Please set it in _config.yml.

Comments

You forgot to set the shortname for Disqus. Please set it in _config.yml.