1. 题目
输入一棵二叉树的根节点,判断该树是不是平衡二叉树。如果某二叉树中任意节点的左右子树的深度相差不超过1,那么它就是一棵平衡二叉树。
2. 解题思路
详情见 面试题55 - II. 平衡二叉树(从底至顶、从顶至底,清晰图解)
2.0 递归
整棵树是平衡二叉树,则每一个子树都是平衡二叉树,很明显可以使用递归。
3. 代码实现
3.0 递归
# Definition for a binary tree node.
# class TreeNode:
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
class Solution:
def isBalanced(self, root: TreeNode) -> bool:
"""
递归,递归判断两棵子树是否为平衡二叉树
"""
def recur(root):
if not root: return 0, True # 返回二叉树的深度
l, l_bal = recur(root.left) # 左子树的深度
r, r_bal = recur(root.right) # 右子树的深度
depth = l+1 if l > r else r+1 # 树的深度
bal = l-r <= 1 if l > r else r-l <= 1
return depth, bal and l_bal and r_bal # 当前树,左子树,右子树是否均平衡
_, flag = recur(root)
return flag
4. 总结
对于二叉树而言,递归比较好些,难的是迭代的咋写,俺不会呀、
5. 参考文献
[1] 剑指offer丛书
[2] 剑指Offer——名企面试官精讲典型编程题
[3] 面试题55 - II. 平衡二叉树(从底至顶、从顶至底,清晰图解)