思路,递归 平衡二叉树就是左右子树的高度差不超过1,与此同时,子节点的左右子树的高度差也不超过1。所以,先判断根节点的左右子树符不符合平衡二叉树的要求,如果符合,则递归判断左右子树是否也都符合,如果不符合,直接返回False
# -*- coding:utf-8 -*-
# class TreeNode:
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
class Solution:
def IsBalanced_Solution(self, pRoot):
# write code here
if not pRoot:
return True
left = self.depth(pRoot.left)
right = self.depth(pRoot.right)
if abs(left-right)>1:
return False
return all([self.IsBalanced_Solution(pRoot.left),self.IsBalanced_Solution(pRoot.right)])
def depth(self,pRoot):
if not pRoot:
return 0
left = right = 0
if pRoot.left:
left = self.depth(pRoot.left)
if pRoot.right:
right = self.depth(pRoot.right)
return max([left,right])+1
京公网安备 11010502036488号