剑指offer——平衡二叉树（python代码，每个节点只遍历一遍）

剑指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