平衡二叉树(二叉树深度+DFS)

题目描述

输入一棵二叉树，判断该二叉树是否是平衡二叉树。

平衡二叉树：又称AVL树，

具有如下性质：

它是一棵空树或它的左右两个子树的高度差的绝对值不超过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 is None:
        	return True

        def DFS(root):

        	# 如果说当前结点为None，返回0
        	if not root:
        		return 0
        	# 返回左子树和右子树的最大值加1
        	return max(DFS(root.right), DFS(root.left)) + 1

        h1 = DFS(pRoot.right)
        h2 = DFS(pRoot.left)
        # 如果说不满足平衡二叉树的性质，返回False
        if abs(h1 - h2) > 1:
        	return False
        # 继续判断左子树和右子树是否满足二叉树的性质
        return self.IsBalanced_Solution(pRoot.right) and self.IsBalanced_Solution(pRoot.left)