输入一棵二叉树的根节点,判断该树是不是平衡二叉树。如果某二叉树中任意节点的左右子树的深度相差不超过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:需要重复遍历节点多次的解法(简单)
在遍历树的每个节点的时候,调用函数height得到它的左右子树的深度。如果每个节点左右子树的深度相差不超过1,那么按照定义它就是一课平衡二叉树,这种思路对应的代码如下:
# 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:
if not root: return True
if (abs(self.height(root.left) - self.height(root.right)) > 1):
return False
else:
return self.isBalanced(root.left) and self.isBalanced(root.right)
def height(self,root):
if not root: return 0
return 1+max(self.height(root.left),self.height(root.right))
思路2:每个节点只遍历一遍(后序遍历)
使用后序遍历二叉树的每个结点,那么在遍历到一个节点之前就已经遍历了它的左右子树。只要在遍历每个节点的时候记录它的深度(某一节点的深度等于它到叶节点的路径的长度),就可以一边遍历,一边判断每个节点是不是平衡的。这种思路的代码如下:
# Definition for a binary tree node.
# class TreeNode:
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
class Solution:
def __init__(self):
self.Balanced = True
def isBalanced(self, root: TreeNode) -> bool:
if not root: return True
pdeth=0
self.LRD_depth(root,pdeth)
return self.Balanced
def LRD_depth(self,root,pdeth):
left_deth=0
if root.left:
left_deth = self.LRD_depth(root.left, pdeth+1)
right_deth=0
if root.right:
right_deth = self.LRD_depth(root.right, pdeth+1)
if abs(left_deth-right_deth)>1:
self.Balanced=False
return max(left_deth,right_deth)+1
来源:力扣(LeetCode)