【剑指Offer】39.平衡二叉树

题目描述

输入一棵二叉树的根结点，判断该树是不是平衡二叉树。如果某二叉树中任意结点的左右子树的深度相差不超过1，那么它就是一棵平衡二叉树。

解题思路

简单解法

遍历树的每个结点时候，调研函数TreeDepth得到它的左右子树的深度。如果每个结点的左右子树的深度相差不超过1，那就是平衡二叉树

代码简洁，但是同一个结点会被重复遍历多次，时间效率不高。

需要寻找一个不需要重复遍历的算法

每个结点只遍历一次的解法

用后序遍历的方式二叉树的每一个结点，在遍历到一个结点之前就已经遍历了它的左右子树，只需要在遍历每个结点的时候记录它的深度，就可以一边遍历一边判断每个结点是不是平衡的

# -*- 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 pRoot == None:
            return True
        if abs(self.TreeDepth(pRoot.left)-self.TreeDepth(pRoot.right))>1:
            return False
        return self.IsBalanced_Solution(pRoot.left) and self.IsBalanced_Solution(pRoot.right)
        
    def TreeDepth(self,pRoot):
        if pRoot == None:
            return 0
        left = self.TreeDepth(pRoot.left)
        right = self.TreeDepth(pRoot.right)
        return max(left+1,right+1)