剑指offer面试题55:
题目描述
输入一棵二叉树,判断该二叉树是否是平衡二叉树。
# -*- 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
# 思路二:每个节点只遍历一次的解法,后续遍历每个节点
depth = [0]
# 想通过该递归函数判断该树是不是平衡二叉树,同时还需要获取左右子树的深度,因此还需要传入引用呢
return self.IsBalanced(pRoot, depth) # 第二个参数要传入引用
def IsBalanced(self, pRoot, dep):
if not pRoot: # 为None,表示空节点,深度为0,也算平衡二叉树
dep[0] = 0
return True
left, right = [0], [0]
if self.IsBalanced(pRoot.left, left) and self.IsBalanced(pRoot.right, right):
if abs(left[0]-right[0]) <= 1:
dep[0] = left[0]+1 if left[0] > right[0] else right[0] + 1
return True
return False