首页 > 其他 > 详细

剑指Offer——II平衡二叉树

时间:2020-08-13 12:27:22      阅读:44      评论:0      收藏:0      [点我收藏+]
class TreeNode:
def __init__(self, x):
self.val = x
self.left = None
self.right = None

# 这道题使用中序遍历加上剪枝的方法来做的。
class Solution:
def isBalanced(self, root: TreeNode) -> bool:
if self.dfs(root) == -1:return False
else:return True
def dfs(self,root):
# 如果节点为空就返回0
if not root:return 0
# 遍历左节点
left = self.dfs(root.left)
# 如果左节点不是平衡二叉树,直接返回
if left == -1:return -1
# 遍历右节点,同理
right = self.dfs(root.right)
if right == -1:return -1
# 这里就保证了左右节点都是平衡二叉树,所以就判断当前root是不是平衡二叉树。
return max(left,right) + 1 if abs(left - right) <= 1 else -1

剑指Offer——II平衡二叉树

原文:https://www.cnblogs.com/cong12586/p/13495402.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!