剑指off 输入一棵二叉树,判断该二叉树是否是平衡二叉树。

法一:遍历每一个点,判断左右子树是否为平衡树   存在大量重复高度的计算,可优化

# -*- 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
        left=self.get_deepth(pRoot.left)
        right=self.get_deepth(pRoot.right)
        if abs(left-right)<2:return self.IsBalanced_Solution(pRoot.left)and self.IsBalanced_Solution(pRoot.right)
        else:return False
    #求树深
    def get_deepth(self,pRoot):
        if pRoot is None:return 0
        left=self.get_deepth(pRoot.left)
        right=self.get_deepth(pRoot.right)
        return max(left,right)+1

法二:对法一进行减枝,当递归时发现已经存在不平衡情况时,不在继续向下递归。设置全局变量boolean进行标记

# -*- coding:utf-8 -*-
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None
class Solution:
    res=True
    def IsBalanced_Solution(self, pRoot):
        # write code here
        if pRoot is None:return True
        self.get_deepth(pRoot)
        return self.res
    #求树深
    def get_deepth(self,pRoot):
        if pRoot is None:return 0
        # 若已经存在非平衡的情况,不再向下递归,返回0   剪纸
        if not self.res:return 0
        left=1+self.get_deepth(pRoot.left)
        right=1+self.get_deepth(pRoot.right)
        if abs(left-right)>1:
            self.res=False
        return max(left,right)

 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值