python判斷平衡二叉樹


題目:輸入一棵二叉樹,判斷該二叉樹是否是平衡二叉樹。若左右子樹深度差不超過1則為一顆平衡二叉樹。
思路:

  1. 使用獲取二叉樹深度的方法來獲取左右子樹的深度
  2. 左右深度相減,若大於1返回False
  3. 通過遞歸對每個節點進行判斷,若全部均未返回False,則返回True
class TreeNode():
    def __init__(self,x):
        self.val  = x
        self.left = None
        self.right = None
class Solution:
    def getDeepth(self,Root):
        if Root is None:
            return 0
        nright = self.getDeepth(Root.right)
        nleft = self.getDeepth(Root.left)
        return max(nright,nleft)+1
    def IsBalance_solution(self,pRoot):
        if pRoot is None:
            return True
        right = self.getDeepth(pRoot.right)
        left = self.getDeepth(pRoot.left)
        if abs(right - left) > 1:
            return False
        return self.IsBalance_solution(pRoot.right) and self.IsBalance_solution(pRoot.left)

 


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM